注:第一次使用NAIDIA显卡的笔记本,要使用pytorch做一些东西,过程可谓是十分艰难,先是CUDA安装不对,再是显卡驱动版本太老,还有pytorch版本与CUDA和显卡驱动不对应导致识别不出显卡,等等问题,因此我决定把我遇到的问题描述出来并给出解决办法。
CUDA安装教程参考如下博文:
【CUDA安装】
【CUDA卸载与安装】
python 执行 torch.cuda.device_count() 返回 0
但是笔记本上是有一张 NVIDIA 3070 GPU,通过下列方式查看笔记本里的 GPU 信息:
问题原因:CUDA、显卡驱动、pytorch、pytorchvision版本不对应
CUDA版本:11.3.1
显卡驱动:最高版本
CUDA11.3 安装 pytorch1.10 教程:点击前往
pytorch下载地址:https://pytorch.org/get-started/locally/
注:博主使用上面官网提供的安装命令进行安装的,输入这个命令回车之后,会等一两分钟才会有反应,所以要等一会才会出现下载的进度条。
最终结果(显示 GPU 数量为1):
torch.cuda.device_count()
查看是否有可用GPU(显示为 True 表示有可用GPU):
torch.cuda.is_available()



