Windows11下pytorch深度学习环境配置(cuda显卡问题相关)

环境:Torch1.10.0+cuda11.3+torchvision0.10.0

1 PyTorch+CUDA安装问题

如果电脑本来装了cuda,那么直接pip安装torch也不会自动配上cuda, 会安装成cpu版的。
如果要安装特定版本的pytorch,我建议是直接到这个链接torch_stable去下载whl文件,然后直接pip install (目录文件名).whl
我下载的是这个,cuda对应11.3,torch是1.10.0
在这里插入图片描述
我发现一件神奇的事情是,他有的版本下载速度快,有的版本速度慢,这也是挺迷的。

另外,直接下载whl文件的好处是,方便挂梯子下载,如果安装失败或者错误方便重来,并且在一开始就已经确定好了cuda的版本和torch的版本。在官网给的命令总是容易出问题。

torchvision和torchaudio也通过这种方式下载whl文件即可。

2 NVIDIA卸载问题

如果要卸载或者更换cuda版本,要看清楚卸载哪些东西。我之前是看了一个博主(报错:Torch not compiled with CUDA enabled看这一篇就足够了)说,要把所有NVIDIA相关的全部卸载。我的建议是不要这样。因为会带着驱动一起卸载了,下面图片标的是驱动相关的,这些就不要卸载了。其他的是安装cuda的时候产生的, 应该可以卸载。
在这里插入图片描述
如果全卸载了,那么电脑所有显卡相关的东西都被移除了,直接导致的就是显卡无法使用了,我记得我刚全卸载完,扩展屏就不显示了。

此外,如果在命令行输入nvidia-smi也报错,说没有该命令,那么大概率就是驱动被卸载了。
在这里插入图片描述
再去官网安装对应的显卡驱动就好,安装完就好了

在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值