!避雷:print(torch.cuda.is_available())返回False

!!!如果torch.cuda.is_available()返回false

如果你遇到这个问题,不用担心你的步骤是不是错了,因为你被conda镜像安装给坑了。你以为下载的是GPU版本,其实镜像下载的是cpu版本,你必须手动去官网用pip下载安装。

1、先查看cuda\cudnn版本是否都对。

执行nvcc --version

 是cuda11.6版本。

2、查看该虚拟环境下安装的pytorch\torchvision版本是否对应cuda11.6。

显然,出了问题。

能看到,安装的torch是cpu版本。

如果你的CUDA,cuDNN版本都对,只有Pytorch安装成了CPU的,那么先对它进行卸载,然后再安装。

卸载方法:删除该虚拟环境的lib->site-packages中的torch、torchaudio和torchvision的六个文件,

 再重新安装正确的torch及torch vision。

安装的步骤见上一篇:二、安装cuda与pytorch的正确安装安装

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值