pycharm中使用GPU跑程序

pycharm中使用GPU跑程序查看机器上 GPU 情况命令 nvidia smi 功能 显示机器上 gpu 的情况命令 nvidia smi l 功能 定时更新显示机器上 gpu 的情况命令 watch n3nvidia smi 功能 设定刷新时间 秒 显示 GPU 使用情况其中左上侧有 0 1 2 3 的编号

查看机器上GPU情况

命令: nvidia-smi

功能:显示机器上gpu的情况

命令: nvidia-smi -l

功能:定时更新显示机器上gpu的情况

命令:watch -n 3 nvidia-smi

功能:设定刷新时间(秒)显示GPU使用情况

pycharm中使用GPU跑程序

其中左上侧有0、1、2、3的编号,表示GPU的编号,在后面指定GPU时需要使用这个编号。

在终端执行程序时指定GPU

CUDA_VISIBLE_DEVICES=1 python your_file.py

这样在跑你的网络之前,告诉程序只能看到1号GPU,其他的GPU它不可见

可用的形式如下:

CUDA_VISIBLE_DEVICES=”” No GPU will be visible

以上内容我也没看明白!,,,,然后我就试试了下边的把代码粘在python程序的开头!然后就可以调用gpu了,不过我已经设置过gpu了,不知道为什么测rbf.nn这个神经网络代码时候,没调用我的gpu!

在Python代码中指定GPU

import os

设置定量的GPU使用量

设置最小的GPU使用量

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/173913.html原文链接:https://javaforall.net

(0)
上一篇 2026年3月27日 上午7:41
下一篇 2026年3月27日 上午7:41


相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号