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

作者：lpred | 2024-01-31 20:07:40

踩

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

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

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

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

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

进入geforce驱动官网

4.开始安装cuda

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/51955