BlueskyFR/conda-cuda-env.md

## conda-cuda-env.md

      
    Raw
  

              conda-cuda-env.md
            
          
    Conda CUDA environnement that takes precedence over the host's

The following is a code snippet that installs torch along with a full cuda environment that takes precedence over the host's.
The only problem is that $TORCH_VERSION must be compatible with/compiled against $CUDA_VERSION, and the latter should be available in conda's nvidia channel (older CUDA versions are not available).
ENV_NAME=segformer CUDA_VERSION=11.1 TORCH_VERSION=1.9 cli=mamba
$cli deactivate
$cli env remove -n $ENV_NAME
$cli create -n $ENV_NAME -c conda-forge -c nvidia python=3.10 gcc=11 gxx cuda-runtime=$CUDA_VERSION cuda-nvcc=$CUDA_VERSION cudatoolkit=$CUDA_VERSION cuda-libraries=$CUDA_VERSION cuda-libraries-dev=$CUDA_VERSION cuda-cudart=$CUDA_VERSION cudnn && \
$cli activate $ENV_NAME && \
pip install light-the-torch && \
ltt install torch==$TORCH_VERSION torchvision

$cli env config vars set CUDA_HOME="/opt/mambaforge/envs/$CONDA_PREFIX"
$cli deactivate
$cli activate $ENV_NAME