NVIDIA GPUs

Most of the frameworks have been build with a specific CUDA version. Please refer to PyTorch or TensorFlow to find the correct CUDA version.

FAQ

SOL tells me that support for NVIDIA is not available?
This is usually caused by a version of SOL without NVIDIA support. Please check if pip3 list installed | grep nec-sol-device-nvidia shows that it is installed, and has the same version as pip3 list installed | grep nec-sol

Which GPUs are supported?
In general all CUDA capable GPU starting from the Kepler architecture (i.e. Tesla K40) are supported.

SOL does not load CUDNN.
We do not bundle CUDNN with SOL. Usually it is already installed within your system. Please manually download it from https://developer.nvidia.com/cudnn (requires a free CUDA developer account) or by running pip3 install nvidia-cudnn-cuXX (replace XX with 11 or 12 depending on your CUDA version).