Skip to content

Can't_install_GPU_packages_from_conda-forge.md

Latest commit

 

History

History
15 lines (12 loc) · 622 Bytes

Can't_install_GPU_packages_from_conda-forge.md

File metadata and controls

15 lines (12 loc) · 622 Bytes
Raw

Can't install GPU packages from conda-forge

Problem

When installing or updating packages from the conda-forge channel, conda refuses to install the GPU version regardless of what's specified.

Solution

Add CONDA_OVERRIDE_CUDA=11.2 before the conda command. For example,

CONDA_OVERRIDE_CUDA=11.2 conda install jax -c conda-forge

This problem occurs because conda makes a virtual package for the CUDA version it detects.

source