RTK icon indicating copy to clipboard operation
RTK copied to clipboard

Investigate dependency to Nvidia packages for Cuda wheels

Open SimonRit opened this issue 2 years ago • 0 comments

According to the Nvidia Cuda documentation, Nvidia provides wheels for Windows and Linux. The following command lines install the required libraries (which are not packaged by RTK, see here)

python3 -m pip install --upgrade setuptools pip wheel
python3 -m pip install nvidia-pyindex
python3 -m pip install nvidia-cuda-runtime-cu11 nvidia-cufft-cu11 nvidia-cublas-cu11

However, the libraries are not put in the system path by conda so I am not sure how to use these wheels. When importing RTK in Python, I still get the error ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory. It's not clear to me if and how we could use these packages...

SimonRit avatar Oct 20 '22 12:10 SimonRit