安装pytorch参考:Python深度学习:安装Anaconda、PyTorch(GPU版)库与PyCharm_哔哩哔哩_bilibili
解决问题参考:Windows 11系统更新CUDA并安装PyTorch环境_win 11 升级 cuda version-优快云博客
在安装pytorch最后的验证步骤时输入import torch,提示UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu.
搞了很久都没搞好。后面看了上面那篇博客才解决问题。
有可能是cuda版本不匹配,注意:可以输入nvidia-smi和nvcc -V查看自己cuda版本,其中前者的要大于后者的才行。否则就更新驱动版本,进入NVIDIA官网:下载 NVIDIA 官方驱动 | NVIDIA
在任务管理器-性能-GPU1看自己的GPU,进官网对比下载最新的版本,一直点下一步。
下载完成以后查看版本
12.0新一些就是ok了,再次尝试import torch
都是ok了。关键在于cuda版本匹配问题。