how to create dockerfile for cuada gpu not cpu
deo999 opened this issue · 1 comments
deo999 commented
hey i am getting dockerfile for cpu for any require python library but how to use it for cuda docker image please tell
python generate.py Dockerfile pytorch lasagne
ufoym commented
python ../generator/generate.py ../docker/Dockerfile pytorch lasagne python==3.6 --cuda-ver 10.0 --cudnn-ver 7
see https://github.com/ufoym/deepo/blob/master/scripts/gen-docker.sh for more examples.