【环境配置】安装了pytorch但是报错torch.cuda.is_availabel()=Flase

解决思路:import torch正常,说明torch包安装正常,但是不能和gpu正常互动,猜测还是pytroch和cuda的配合问题

1.查看torch包所需的cuda版本

我的torch是2.0.1,在现在是比较新的包,需要12以上的cuda支持,我用nvcc -V或者set cuda查看当前环境路径下的cuda版本,显示为11.1,这显然无法匹配

2.匹配cuda版本

那就需要更新了,首先检查我的硬件是否支持新版本的cuda,nvidia-smi,可以看到我的显卡最高支持到12.3,因此就去官网下载CUDA Toolkit 12.3 Update 2 下载 |NVIDIA 开发者

但是安装前这时候还要看cuda和geforce驱动匹配问题

3.匹配geforce驱动

cuda和geforce驱动的匹配关系,可以看到,如果想安装cuda12.1,至少匹配到527.41

CUDA 12.3 Update 2 Release Notes (nvidia.com)

进入geforce驱动官网

NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

安装Pytorch时NVIDIA驱动更新,CUDA版本问题_为什么pytroch不更新amd驱动-CSDN博客

4.开始安装cuda

这网页进来 version没有12 你别管 直接看大标题对就下

CUDA Toolkit Archive | NVIDIA Developer

相关推荐
默 语14 分钟前
AI驱动软件测试全流程自动化:从理论到实践的深度探索
运维·人工智能·驱动开发·ai·自动化·ai技术·测试全流程
说私域20 分钟前
基于开源AI大模型、AI智能名片与S2B2C商城小程序的购物中心精准零售数据架构研究
人工智能·小程序·开源
key0628 分钟前
大模型在网络安全领域的应用与评测
网络·人工智能·web安全
北京耐用通信1 小时前
破解工业通信瓶颈:耐达讯自动化Modbus转Profinet网关连接驱动器的奥秘
人工智能·物联网·网络协议·自动化·信息与通信
应用市场1 小时前
OpenCV进阶:图像变换、增强与特征检测实战
人工智能·opencv·计算机视觉
Moniane1 小时前
Python爬虫入门:从零到数据采集
深度学习
说私域1 小时前
开源链动2+1模式、AI智能名片与S2B2C商城小程序:社群经济的数字化重构路径
人工智能·小程序·开源
lingchen19061 小时前
卷积神经网络中的卷积运算原理
深度学习·计算机视觉·cnn
rengang661 小时前
智能化的重构建议:大模型分析代码结构,提出可读性和性能优化建议
人工智能·性能优化·重构·ai编程
灵遁者书籍作品1 小时前
语言的拓扑学约束公理:语言对实在的描述具有拓扑不变量——某些真理必须通过悖论、沉默或隐喻表达
人工智能·计算机视觉