torch.version.cuda 没反应(cuda与torch的安装匹配)
此博客主要用于记录个人的问题解决 。如果能帮到路过的朋友那再好不过啦 。
我在某网站的评论所分享的链接下载的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版本 。
但在控制面板显示11.3
我不是很了解为何会有这样的矛盾出现 。如果有懂的大佬麻烦帮我科普一下。谢谢乐!!!!
我开始弄不明白了 ,所以暂时开摆 。试试换torch
方法二 、换torch
在网上冲浪搜索后发现有大佬说有的torch本身不支持cuda 。所以我想我下载的torch也有这个可能。所以去趟官网下载torch 。(之前是担心下载速度过慢 ,所有用的分享来的源下载)
这次下载安装很顺利 ,速度很快,一小会就好了 。但是在红线处死机了一到二分钟 。
(为什么会有这种原因 。我不是很了解下面的warning的“读取超时 ”是何意思 。如果有懂的大佬麻烦帮我科普一下 。谢谢乐!!!!)
安装过程中会自动替换掉以前的1.10.1torch
之后就是再尝试 。发现尝试成功了 。
以上为尝试的代码 。(抄的李沐大哥)
总结:没换cuda 。换了torch 。换的是可支持cuda的torch版本 。
创心域SEO版权声明:以上内容作者已申请原创保护,未经允许不得转载,侵权必究!授权事宜、对本内容有异议或投诉,敬请联系网站管理员,我们将尽快回复您,谢谢合作!