当你的系统拥有多个版本cuda时,想切换你想用的版本的时,有两种方法
方法1:临时修改,在cmd命令行进行以下操作
set CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8"
set PATH=%CUDA_HOME%\bin;%CUDA_HOME%\libnvvp;%PATH%
nvcc --version
方法2:环境变量只保留你想要的那个版本
如何查看当前的cuda版本
比较文档的cuda与pytorch
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
export CUDA_HOME=/usr/local/cuda-12.1
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
评论