无权限

Derrick博客站

如何切换cuda版本

当你的系统拥有多个版本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版本

nvcc --version

比较文档的cuda与pytorch

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118 

评论

快捷导航

把好文章收藏到微信

打开微信,扫码查看

关闭

还没有账号?立即注册