Hello pmyc3 experts
I’m trying to run pymc3 with GPU support with following setup:
ubuntu-20.04, kernel 5.4.0-72-generic
I created a conda environment and installed:
pip install pymc-3.11.0 conda install -c conda-forge pygpu
I installed cuda and cudnn from the nvidia site. I can run nvidida-smi and it detects my NVIDIA GeForce GTX 1080
I set up a .theanorc file in my home dir:
[global] floatX = float32 device = cuda0 force_device = True optimizer_including=cudnn [dnn] include_path=/usr/local/cuda-10.0/include library_path=/usr/local/cuda-10.0/lib64 [cuda] root=/usr/local/cuda-10.0/
But when I import pymc3 I get the following:
/home/hadron/myutils/anaconda3/envs/myenv1/lib/python3.8/site-packages/theano/gpuarray/dnn.py:192: 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( Using cuDNN version 7401 on context None Mapped name None to device cuda0: NVIDIA GeForce GTX 1080 (0000:01:00.0) ERROR (theano.gpuarray): Could not initialize pygpu, support disabled Traceback (most recent call last): File "/home/hadron/myutils/anaconda3/envs/myenv1/lib/python3.8/site-packages/theano/gpuarray/__init__.py", line 262, in <module> use(config.device) File "/home/hadron/myutils/anaconda3/envs/myenv1/lib/python3.8/site-packages/theano/gpuarray/__init__.py", line 251, in use optdb.add_tags("gpuarray_opt", "fast_run", "fast_compile") AttributeError: module 'theano.gpuarray.optdb' has no attribute 'add_tags'
As you can see I used cuda 10.0 and installed cuDNN 22.214.171.124, but I also tried with others, i.e. more recent versions (starting from cuda 11.3 and working my way down), trying to address the message “Your cuDNN version is more recent than Theano” but that still doesn’t work.
Can anyone help me with that? I’m quite stuck with it