UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu.

安装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版本匹配问题。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值