此博客主要用于记录个人的问题解决。如果能帮到路过的朋友那再好不过啦。
我在某网站的评论所分享的链接下载的torch,似乎是阿里的源。是1.10.1的torch。
pip show torch#查看torch版本
后再下载的CUDA,下载到了11.6版本的。后试运行,发现torch与CUDA无法匹配使用。
接下来我尝试了两个方法处理。
#我后来借用方法二解决了问题。方法一并没有解决我的问题,希望懂的朋友能解答一下我的疑问
方法一、改CUDA
我认为是cuda的版本过高,导致的torch与cuda的不可匹配。于是尝试卸载重装cuda11.3。安装多次后发现,难以卸载cuda11.6,反复卸载11.6再安装11.3,最终用代码“nvidia-smi”查看版本,却依旧显示是11.6版本。
我不是很了解为何会有这样的矛盾出现。如果有懂的大佬麻烦帮我科普一下。谢谢乐!!!!
我开始弄不明白了,所以暂时开摆。试试换torch
方法二、换torch
在网上冲浪搜索后发现有大佬说有的torch本身不支持cuda。所以我想我下载的torch也有这个可能。所以去趟官网下载torch。(之前是担心下载速度过慢,所有用的分享来的源下载)
安装过程中会自动替换掉以前的1.10.1torch
之后就是再尝试。发现尝试成功了。
以上为尝试的代码。(抄的李沐大哥)
总结:没换cuda。换了torch。换的是可支持cuda的torch版本。