概述

由于跑项目发现需要配置不同版本的Pytorch,而不同版本的Pytorch又对应不同版本的CUDA,于是有了在Win上装多个CUDA的打算

默认已经在电脑上装了一个CUDA

屏幕截图 2023-09-30 112423

现在开始下载第二个CUDA版本,前面下载的操作和普通安装的几乎一样

安装CUDA

CUDA下载链接

下载自己需要的CUDA版本,以我自己要安装的为例

屏幕截图 2023-09-30 095614

屏幕截图 2023-09-30 095815

下载打开exe文件进行安装,第一步不用管,直接点击确认

屏幕截图 2023-09-30 100402

后面选择自定义安装,只选择安装CUDA

屏幕截图 2023-09-30 100534

其他一路下一步就可以了

安装cuDNN

cuDNN下载链接

PS:下载需要登录一下账号

屏幕截图 2023-09-30 100047

下载下来后,把压缩包解压,得到如下文件

image-20230930100746137

bin,include文件夹中的文件,分别复制到下列地址对应的文件夹下(CUDA的安装地址,以我的地址为例),遇到提示,为全部选择覆盖和替换。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7

需要注意的是,lib文件夹里的所有文件,需要复制到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\lib\x64

至此,安装工作就做完了,可以看到电脑上已经有两个CUDA版本了

image-20230930101157014

配置环境

由于CUDA11.7是后面安装的,所以可以看到系统把之前的环境覆盖了

屏幕截图 2023-09-30 103109

以作者在为例,系统变量的PATH里配置一下路径

CUDA_11.7

%CUDA_PATH_V11_7%\lib\x64
%CUDA_PATH_V11_7%\include
%CUDA_PATH_V11_7%\extras\CUPTI\lib64
%CUDA_PATH_V11_7%\bin
%CUDA_PATH_V11_7%\libnvvp

CUDA_11.0

%CUDA_PATH_V11_0%\lib\x64
%CUDA_PATH_V11_0%\include
%CUDA_PATH_V11_0%\extras\CUPTI\lib64
%CUDA_PATH_V11_0%\bin
%CUDA_PATH_V11_0%\libnvvp

配置好如下:

微信截图_20230930105036

配置好之后,默认CUDA版本是11.7的,如果要切换到11.0版本的,只需要把下面的五条整体上移到CUDA_11.7配置的前面

微信截图_20230930105411

验证

测试CUDA11.7

首先验证CUDA_11.7是否配置成功

image-20230930105601201

测试Pytorch是否可以调用显卡

python
import torch
torch.cuda.is_available()
torch.cuda.get_device_name(0)
torch.version.cuda

image-20230930120418607

测试CUDA11.0

先把系统变量里Path里相关路径上移

微信截图_20230930110409

然后重启!重启!重启!

验证CUDA_11.7是否配置成功

image-20230930112425329

测试Pytorch是否可以调用显卡

image-20230930112626663

总结

切换CUDA的时候除了要修改系统变量Path中的路径顺序,一定要记得重启!