电脑基础 · 2023年3月31日

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。(之前是担心下载速度过慢,所有用的分享来的源下载)

cuda与torch的安装匹配

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

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

cuda与torch的安装匹配

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

cuda与torch的安装匹配

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

cuda与torch的安装匹配

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

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