cudatoolkit
package. Other packages such as Numba do not have a cudatoolkit
dependency, because they can be used without the GPU.
Installing Linux on Windows with WSL
cudatoolkit
conda package in their environment. To select a cudatoolkit
version, add a selector such as cudatoolkit=12.1
to the version specification.
Required NVIDIA driver versions, excerpted from the NVIDIA CUDA Toolkit Release Notes:
CUDA Version | Linux x86_64 Driver Version | Windows x86_64 Driver Version |
---|---|---|
CUDA 11.8.x | >= 450.80.02 | >= 452.39 |
CUDA 12.0.x | >= 525.60.13 | >= 527.41 |
CUDA 12.1.x | >= 525.60.13 | >= 527.41 |
CUDA 12.2.x | >= 525.60.13 | >= 527.41 |
tensorflow
package in Anaconda, one of which is the NVIDIA GPU version. This is selected by installing the meta-package tensorflow-gpu
.
tensorflow
package name and will use whatever version of TensorFlow is installed. This makes it straightforward to switch between variants in an environment.
mxnet-gpu
meta-package.
cudatoolkit
package is present.: