先打开cmd控制面板,输入nvcc --version,确认电脑上没有cuda
打开nvidia控制面板,看到显卡型号是3050
在帮助–系统信息下打开弹窗,在“组件”下找到CUDA版本,可以看到要下载的是11.4.1版本的
打开
nvidia的官方cuda下载地址
,找到11.4.1下载
选择深绿色的部分,开始下载
下载后按照默认设置安装即可,其中解压地址可以随便选,安装完之后会把解压的中间文件删除。
重启之后,在cmd中输入命令nvcc --version,发现安装成功
接下来吧cuda添加进环境变量里,用windows自带的搜索可以打开
在“系统变量”下选择“path”,然后点编辑
添加以下两条:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
这样,就可以正常使用CUDA了
报错:RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
报错:RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1原因:在使用Pytorch加载模型时报错。加载的模型是用两个GPU训练的,而加载模型的电脑只有一个 ...