起因是在python3里执行下述代码,发现是false,于是想想怎么解决。
import torch
torch.cuda.is_available()
首先搜到在windows中安装cuda的指引,发现主要步骤为:
1. 配平显卡驱动、cuda版本、pytorch版本。
我是pytorch 1.11.0装好了才发现这个问题,根据前车之鉴,我选择把显卡驱动升级为最新版本512.95,并开始准备安装1.11.0的cuda。
2023年4月24日:补充具体步骤
1. 在Previous PyTorch Versions | PyTorch网址中找自己已经装了的pytorch版本号(如1.8.0)
注意,如果之前已经安装了(像我就错装了cpu版本的torch),必须要先卸载(最好pip和conda都uninstall一下),不然就会遇到问题:Key already registered with the same priority: GroupSpatialSoftmax,重装解决一切问题。
按照提示安装到conda环境里:
安好后验证一下:
至于nvcc -c的使用:
注意
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
这里的cudatoolkit和NVIDIA的不是同一个东西,两者都需要安装。
2. 安装cuda
参考官方安装教程以及这个博客的第三步和第四步,下载win对应版本的Installer,全部安装后验证PASS。
注意:安装包2.5G,从安装解压到结束至少需要13G的空间,最终安装好的大小为6G左右
3. 安装cuDNN
参考官方安装教程,找到对应的版本下载后解压。
4. 验证
5. GPU不起作用
可能遇到的问题:
pytorch和cudatoolkit版本严格对应
参考文章:
cuda与显卡驱动版本对照
全流程安装步骤