位置: IT常识 - 正文
cuda与torch的安装匹配(cuda torch)
编辑：rootadmin推荐整理分享cuda与torch的安装匹配(cuda torch)，希望有所帮助，仅作参考，欢迎阅读内容。
文章相关热门搜索词:torch.cuda.set_device,torch. cuda. is_available,torch.version.cuda,torch.cuda.is available,torch.version.cuda,torch.cuda.set_device,torch.cuda.set_device,torch.version.cuda,内容如对您有帮助，希望把文章链接给更多的朋友！
此博客主要用于记录个人的问题解决。如果能帮到路过的朋友那再好不过啦。
我在某网站的评论所分享的链接下载的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版本。
上一篇：vue3：获取当前路由地址(vue3获取当前vue实例)
下一篇：在ROS下使用Cartographer的纯定位模式，并实时获取定位位姿数据(ros call)