Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Pytorch-Encoding
Commits
deec6dea
Unverified
Commit
deec6dea
authored
Aug 01, 2020
by
Hang Zhang
Committed by
GitHub
Aug 01, 2020
Browse files
ADD Docker (#305)
* add docker
parent
e219aa7e
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
23 additions
and
0 deletions
+23
-0
Dockerfile
Dockerfile
+21
-0
scripts/build_docker.sh
scripts/build_docker.sh
+1
-0
scripts/run_docker.sh
scripts/run_docker.sh
+1
-0
No files found.
Dockerfile
0 → 100644
View file @
deec6dea
FROM
nvcr.io/nvidia/pytorch:19.05-py3
# Set working directory # Set working directory
WORKDIR
/workspace
#ENV PYTHONPATH "${PYTHONPATH}:/workspace"
RUN
apt-get update
&&
DEBIAN_FRONTEND
=
noninteractive apt-get
install
-y
python3-tk python-pip git tmux htop tree
RUN
python
-m
pip
install
--upgrade
pip
RUN
python
-m
pip
install
torch
==
1.4.0
RUN
python
-m
pip
install
torchvision
==
0.5.0
RUN
python
-m
pip
install
pycocotools
==
2.0.0
#RUN chmod a+rwx -R /opt/conda/
COPY
./setup.py .
COPY
./encoding ./encoding
RUN
python setup.py develop
COPY
./experiments ./experiments
scripts/build_docker.sh
0 → 100644
View file @
deec6dea
docker build
-t
encoding
.
scripts/run_docker.sh
0 → 100644
View file @
deec6dea
docker run
--rm
-it
--ulimit
memlock
=
-1
--ulimit
stack
=
67108864
-v
$COCO_DIR
:/coco
--ipc
=
host
--gpus
=
all encoding
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment