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

QianMo-WXJ

已于 2023-09-20 16:46:12 修改

阅读量404

点赞数

CC 4.0 BY-SA版权

文章标签：深度学习人工智能机器学习

于 2023-09-20 16:45:38 首次发布

用户在新项目中遇到CUDA可用性判断差异，torch.cuda.is_available()在不同项目中返回值不同。可能需要检查环境配置，确保torch版本与CUDA版本对应。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

问题来源：

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

查看

和

是否对应。

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

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