更新显卡驱动可以安装高版本的cuda
可以通过打开cmd输入nvidia-smi来查看
所以说不高于12.3的版本都可以安装
cuda版本匹配pytorch版本
打开网址:https://pytorch.org/
验证完后,就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌
关键一步
把这个复制进去安装一下
运行
根本原因
pycharm安装的torch只支持cup计算,所以需要重新装过
可以通过打开cmd输入nvidia-smi来查看
所以说不高于12.3的版本都可以安装
打开网址:https://pytorch.org/
验证完后,就可以吭哧吭哧的去下载安装了
搞完后还是发型代码跑出来GPU = 0
不要慌
把这个复制进去安装一下
pycharm安装的torch只支持cup计算,所以需要重新装过