解决Assertion:Error Torch not compiled with CUDA enabled问题(CUDA版本正确)

一、问题说明

用Anaconda安装 PyTorch 2.0.1,GPU版,CUDA版本为11.7:

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

但是torch可用,cuda不可用:

import torch
#torch是否可用;
print(torch.__version__)    #2.0.1

#cuda是否可用;
print(torch.cuda.is_available())    #False

看网上的解决方案,大多需要确定是否安装为GPU版本,以及CUDA版本是否一致(见下一节),但我这都没问题。

二、确定CUDA版本(运行时API,Runtime API)

打开命令提示符(Win+R -> cmd)输入:

nvcc -V

出来的最后两行就是版本号
在这里插入图片描述

网上还有其他方法查看CUDA,但是看的不是一个东西(详见另一篇博客),以这个为准就完事了。

三、解决方案

安装更低版本PyTorch即可(2.0.0及以下),原因没整明白,离谱。
在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值