安装好cuda后解决torch.cuda.device_count() == 0

更新显卡驱动可以安装高版本的cuda

可以通过打开cmd输入nvidia-smi来查看
安装好cuda后解决torch.cuda.device_count() == 0_第1张图片
所以说不高于12.3的版本都可以安装

cuda版本匹配pytorch版本

打开网址:https://pytorch.org/
安装好cuda后解决torch.cuda.device_count() == 0_第2张图片
验证完后,就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌

关键一步

安装好cuda后解决torch.cuda.device_count() == 0_第3张图片
安装好cuda后解决torch.cuda.device_count() == 0_第4张图片
把这个复制进去安装一下

运行

安装好cuda后解决torch.cuda.device_count() == 0_第5张图片

根本原因

pycharm安装的torch只支持cup计算,所以需要重新装过
安装好cuda后解决torch.cuda.device_count() == 0_第6张图片

你可能感兴趣的:(深度学习,深度学习)