Add CUDA 7.5 and 8.0 Dockerfiles #4114

Merged
merged 2 commits into from Dec 7, 2016

Projects

None yet

2 participants

@pineking
Contributor
pineking commented Dec 6, 2016

Fix #4102

@pineking pineking Add CUDA 7.5 and 8.0 Dockerfiles
ae51ff0
docker/cuda/8.0/Dockerfile
+ sed -i 's/USE_CUDNN = 0/USE_CUDNN = 1/g' config.mk && \
+ make -j"$(nproc)"
+
+RUN cd /mxnet/python && python setup.py install
@piiswrong
piiswrong Dec 6, 2016 Member

setting PYTHONPATH is the recommended way of installing mxnet.
~/mxnet is the recommended location

@pineking pineking Use ~/mxnet as the workdir, set PYTHONPATH in dockerfile
97d7b1a
+
+ENV PYTHONPATH /root/mxnet/python
+
+WORKDIR /root/mxnet
@piiswrong
piiswrong Dec 6, 2016 Member

Is this accessible by users if its in /root?

@pineking
pineking Dec 7, 2016 Contributor

sure, 'root' is the default user in docker container. If we can use the command docker run or nvidia-docker run to start a container, then we are granting root privileges in the container.

@piiswrong
piiswrong Dec 7, 2016 Member

what do people typically do for docker? /root or $HOME?

@pineking
pineking Dec 7, 2016 Contributor

'/root' and $HOME are equal in docker container, since the default user is root, and the home dir of root is ‘/root’. so '/root/mxnet' is equal to '~/mxnet'.

@piiswrong piiswrong merged commit 3f6cc9d into dmlc:master Dec 7, 2016

3 checks passed

continuous-integration/appveyor/pr AppVeyor build succeeded
Details
continuous-integration/travis-ci/pr The Travis CI build passed
Details
default Build finished.
Details
@pineking pineking deleted the pineking:update_dockerfile branch Dec 7, 2016
@pineking
Contributor
pineking commented Dec 7, 2016

@piiswrong
I want to create an another PR to update the docker/README.md,since the content of the README.md is out-of-date. The pre-built docker images are used as the example in README.md and they also need to be updated. These images are hosted on the https://hub.docker.com/r/dmlc/mxnet/ , do you have the account information for login and I can push the docker images?

@piiswrong
Member

please push to your own account.

@mli Could you pull from his account and push to dmlc afterwards?

@pineking
Contributor
pineking commented Dec 7, 2016

OK.
I have pushed the docker images here: https://hub.docker.com/r/pineking/mxnet/
so @mli can pull the images and then push them to dmlc or build himself from the Dockerfile directly.

But I think the best choice is to setup an automated build on hub.docker.com https://docs.docker.com/engine/tutorials/dockerrepos/#/automated-builds

please let me know if the docker images are updated, then I can update the docker/README.md

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment