解决思路: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