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版本

cuda与torch的安装匹配

但在 控制面板显示11.3

cuda与torch的安装匹配

我不是很了解为何会有这样的矛盾出现。如果有懂的大佬麻烦帮我科普一下。谢谢乐!!!!

我开始弄不明白了,所以暂时开摆。试试换torch

方法二、换torch

在网上冲浪搜索后发现有大佬说有的torch本身不支持cuda。所以我想我下载的torch也有这个可能。所以去趟官网下载torch。(之前是担心下载速度过慢,所有用的分享来的源下载)

这次下载安装很顺利,速度很快,一小会就好了。但是在红线处死机了一到二分钟。

(为什么会有这种原因。我不是很了解下面的warning的”读取超时”是何意思。如果有懂的大佬麻烦帮我科普一下。谢谢乐!!!!)

cuda与torch的安装匹配

安装过程中会自动替换掉以前的1.10.1torch

cuda与torch的安装匹配

之后就是再尝试。发现尝试成功了。

cuda与torch的安装匹配

以上为尝试的代码。(抄的李沐大哥)

总结:没换cuda。换了torch。换的是可支持cuda的torch版本。

Original: https://blog.csdn.net/qq_59412678/article/details/122566648
Author: 朱哥靓965
Title: cuda与torch的安装匹配

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/706385/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球