'Enable Nvidia GPU CUDA support for theano and PyMC3 for Windows 11
Has anyone figured out how to install GPU support for Theano and PyMC3 on Windows 11? I keep getting an error message about a DLL load failure. Here are the steps I’ve taken:
- Install MSVC C++ Build tools and add to system variable path
- Install latest Nvidia drivers, CUDA toolkit, and added cuDNN .dll files to folders
- Created .theanorc with the following line
[global], device = cuda, floatX = float32and added to base directory - Added 'THEANO_FLAGS' system variable with
floatX=float32,device=cuda(is this redundant with step 3?) - Created the following conda environment:
conda create -n pymc_env_gpu -c conda-forge python libpython mkl-service numba python-graphviz scipy arviz pandas scikit-learn m2w64-toolchain pygpu ipykernel statsmodels
- Activate new env and install pymc3 with
pip install pymc3
The error message: ImportError: DLL load failed while importing m169e842c8a94977b534b5d54311bcacefb2595c2a7bb1124600448dca20d9048: The specified module could not be found.
Any help would be appreciated!
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|
