Closed turian closed 2 years ago
I am using deepo generate on a GPU machine with CUDA 10.2
deepo generate
However, when I run it, the Dockerfile is for CPU not CUDA.
I see that you need to use --cuda-ver 10.2. Could you please update the documentation?
--cuda-ver 10.2
There is also a bug. If you don't have cudnn or pass in this command line arg, you get the following line in Dockerfile:
FROM nvidia/cuda:10.2-cudnnNone-devel-ubuntu18.04
It should instead be:
FROM nvidia/cuda:10.2-devel-ubuntu18.04
I am using
deepo generateon a GPU machine with CUDA 10.2However, when I run it, the Dockerfile is for CPU not CUDA.
I see that you need to use
--cuda-ver 10.2. Could you please update the documentation?There is also a bug. If you don't have cudnn or pass in this command line arg, you get the following line in Dockerfile:
It should instead be: