【已解决】在jupyter里运行torch.cuda.is_available()，显示True，在pycharm中运行却显示false。

在jupyter里运行torch.cuda.is_available()，显示True，在pycharm中运行却显示false。原因在于jupyter 运行环境和pycharm 运行环境不同，前者支持gpu,后者配置不到位。

QuantumYou

716人浏览 · 2024-05-26 20:29:04

QuantumYou · 2024-05-26 20:29:04 发布

文章目录

问题概述
1、在Jupyter中GPU运行true
2、在pycharm中GPU运行false
3、个人解决方案仅供参考

问题概述

在jupyter里运行torch.cuda.is_available()，显示True，在pycharm中运行却显示false。
原因在于jupyter 运行环境和pycharm 运行环境不同，前者支持gpu,后者配置不到位

1、在Jupyter中GPU运行true

在这里插入图片描述

2、在pycharm中GPU运行false

在这里插入图片描述

3、个人解决方案仅供参考

个人的解决方案，仅供参考，仅代表我遇到的问题的摸索过程，也许你遇到的问题与我的有些许不同
如下图在设置中找到项目python解释器，设置anconda 环境和安装项目所需要的软件包(一般可以自动导包)
最后在运行配置中也更新，注意各种版本号（tensorflow-gpu，torch，numpy,cuda ，显卡驱动等）要对应 🔥🔥

相关文章参考2

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..