当前位置:   article > 正文

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

assertionerror: torch not compiled with cuda enabled

一、问题说明

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
  • 1

但是torch可用,cuda不可用:

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

#cuda是否可用;
print(torch.cuda.is_available())    #False
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

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

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

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

nvcc -V
  • 1

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

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

三、解决方案

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/491332
推荐阅读
相关标签
  

闽ICP备14008679号