search

puzzlelib / PuzzleLib

Deep Learning framework with NVIDIA & AMD support
https://docs.puzzlelib.org
Apache License 2.0
59 stars 11 forks source link

Doesn't work with CUDA 11.4 #2

Open EmilPi opened 3 years ago

EmilPi commented 3 years ago

I install PuzzleLib from within PuzzleLib path with command pip3 install ./ --install-option="--backend=cuda" . Installation goes without error. Now, when I run python3 TestLib/CnnCifar10NIN.py, I get

Traceback (most recent call last): File "TestLib/CnnCifar10NIN.py", line 3, in from PuzzleLib.Containers import Sequential File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Containers/init.py", line 1, in from PuzzleLib.Containers.Container import Container File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Containers/Container.py", line 6, in from PuzzleLib.Modules.Module import Module, ModuleError File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Modules/init.py", line 1, in from PuzzleLib.Modules.Activation import Activation, ActivationType, sigmoid, tanh, relu, leakyRelu, elu, softPlus, clip File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Modules/Activation.py", line 7, in from PuzzleLib.Backend import gpuarray File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Backend/gpuarray.py", line 201, in autoinit() File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Backend/gpuarray.py", line 40, in autoinit initCuda() File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Backend/gpuarray.py", line 50, in initCuda from PuzzleLib.Cuda import Backend, Utils File "/home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Cuda/Backend.py", line 6, in from PuzzleLib.Cuda import Driver as CudaDriver ImportError: /home/emil/.local/lib/python3.8/site-packages/PuzzleLib/Cuda/Driver.cpython-38-x86_64-linux-gnu.so: undefined symbol: cudnnSetRNNDescriptor

Is there any workaround? If no, I think docs should mention specific CUDA/CuDNN versions PuzzleLib works with.