torch.cuda.is_available() 在有的项目中返回True有的返回Flase

问题描述,刚下了一个项目,不能用CUDA  同一个环境不同项目中 torch.cuda.is_available() 返回值不同

问题来源:torch.cuda.is_available() 在有的项目中返回True有的返回Flase_第1张图片

这里的运行配置有问题 选择编辑配置并修改对应的解释器

查看torch.cuda.is_available() 在有的项目中返回True有的返回Flase_第2张图片

torch.cuda.is_available() 在有的项目中返回True有的返回Flase_第3张图片是否对应。

import torch
print(torch.__version__)
print(torch.cuda.is_available())

1.13.1+cu116 #在控制台检查输出及运行中对应的torch版本是否为cuda版本
True

你可能感兴趣的:(深度学习,人工智能,机器学习)