Sign inSign up

jermine/dl

By jermine

•Updated over 8 years ago

Dockerfile for DeepLearn (Mxnet , Pytorch , Caffe2 , Tensorflow)

Image
5

5.1K

jermine/dl repository overview

⁠docker-dl

Dockerfile for DeepLearn (Mxnet , Pytorch , Caffe2 , Tensorflow). This repository recommends developers to use, which covers cuda9.1, cudnn7.0, tensorflow1.8, opencv3.4.1, pytorch0.4.0, mxnet1.1 ,caffe,caffe2 .

⁠The docker registry is :

https://hub.docker.com/r/jermine/dl/⁠

⁠The Github repo is:

https://github.com/JermineHu/docker-dl⁠

⁠Used

You can switch the environments by docker images tags!

⁠Checked nvidia driver info
docker run --runtime=nvidia --rm -it jermine/dl:tensorflow.gpu nvidia-smi
⁠Get pip3 list
docker run --rm -it jermine/dl:tensorflow.gpu pip3 list --format=columns

Tag summary

Content type

Image

Digest

Size

964.9 MB

Last updated

over 8 years ago

docker pull jermine/dl