当前位置:   article > 正文

Windows11下pytorch深度学习环境配置(cuda显卡问题相关)_torch==1.11.0,下载cuda11.3只有windows1

torch==1.11.0,下载cuda11.3只有windows1

环境: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也报错,说没有该命令,那么大概率就是驱动被卸载了。
在这里插入图片描述
再去官网安装对应的显卡驱动就好,安装完就好了

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/128625
推荐阅读
相关标签
  

闽ICP备14008679号