深度学习环境配置,遇见torch.cuda.is_available() 输出为Flase的BUG。其中一种情况解决方案

402 阅读1分钟

奋斗一晚上解决BUG记录一下,因为全网几乎没有遇上我这个bug

这个配置BUG的历程记录: 在英伟达官网下载CUDA,默认推荐给我11.8版本,但是torch最高支持到11.7,在配置11.8的时候,配置了11.7的torch(BUG#1)。 卸载CUDA11.8后,重新下载11.7,再配置torch后,就是如下一直FALSE的输出,很折磨人。 后续大佬查出是因为前面配置的torch有冲突,卸载后,重新下载就好,细节如下。 因为需要使用CUDA和Pytorch

import torch
print(torch.cuda.device_count())
#可用gpu数量
print(torch.cuda.is_available()) # #是否可用gpu
print(torch.version.cuda)
print(torch.__version__)

执行上述代码一直是FALSE和不显示cuda的信息 去某平台求助了大神找到解决方案

  1. 终端输入conda activate 你的环境
  2. pip list
  3. 我的终端如下

c12bded5a10530dd6ff0f9b4663ce21.png 然后卸载

fa34d544333d4d02a5747e65f5d7319.png 终端输入pip uninstall torch torchvision torchaudio 一直按y

后面重新在Pytorch.org 下载对应的版本就ok

注意CUDA和torch版本需对应