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

目录

  • 一、问题说明
  • 二、确定CUDA版本(运行时API,Runtime API)
  • 三、解决方案

一、问题说明

用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

出来的最后两行就是版本号
解决Assertion:Error Torch not compiled with CUDA enabled问题(CUDA版本正确)_第1张图片

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

三、解决方案

安装更低版本PyTorch即可(2.0.0及以下),原因没整明白,离谱。
解决Assertion:Error Torch not compiled with CUDA enabled问题(CUDA版本正确)_第2张图片

你可能感兴趣的:(分享,python,pytorch,深度学习,bug)