ML | set CUDA_HOME in a conda environment

set CUDA_HOME in a conda environment

1
2
3
4
5
6
micromamba install -c conda-forge cudatoolkit-dev
which nvcc
nvcc --version
echo $CUDA_HOME
export CUDA_HOME=$CONDA_PREFIX
echo $CUDA_HOME

References

  1. How to set my CUDA_HOME in a conda environment?
  2. https://github.com/conda/conda/issues/7757#issuecomment-967303351
ML
#AI #ML #pytorch #NVIDIA #CUDA
ML | set CUDA_HOME in a conda environment
https://waipangsze.github.io/2025/01/22/ML-set-CUDA_HOME-in-a-conda-environment/
Author
wpsze
Posted on
January 22, 2025
Updated on
January 23, 2025
Licensed under