deepo generate does not find CUDA

[Joseph Turian]

I am using deepo generate on a GPU machine with CUDA 10.2

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?

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