Closed sedghi closed 2 years ago
Sorry for the late reply.
By default, there exists a file /root/.theanorc
inside a deepo image with theano
. And it should be OK to use GPU in this case, though theano
is deprecated now. If you simply import theano
in the latest deepo image, there should be a warning like this:
/usr/local/lib/python3.6/dist-packages/theano/gpuarray/dnn.py:184: UserWarning: Your cuDNN version is more recent than Theano. If you encounter problems, try updating Theano or downgrading cuDNN to a version >= v5 and <= v7.
warnings.warn("Your cuDNN version is more recent than "
Using cuDNN version 8005 on context None
Mapped name None to device cuda0: NVIDIA GeForce RTX 2070 (0000:01:00.0)
We think the problem in your case is probably the lack of /root/.theanorc
, especially when a directory from your host is mounted into the /root
and therefore /root/.theanorc
does not exist any more.
You can, of course, create a /root/.theanorc
again with the following settings:
[global]
device = cuda
floatX=float32
or use theano
in this way
THEANO_FLAGS='device=cuda,floatX=float32' python -c "import theano"
Hi, I've followed instructions of how to run theano gpu using the deepo; unfortuantely i'm not able to run the theano code with gpu. it uses the cpu instead
steps that I took
docker run --gpus all -it ufoym/deepo:theano bash
and I'm running the following test code (from theano documentations)
I'm getting the following output which says it have used cpu instead of gpu:
here is my nvidia-smi inside my docker
any help is appreciated