使用GPU跑程序

绝地灬酷狼 2022-05-13 20:17 261阅读 0赞

查看机器上GPU情况

命令： nvidia-smi

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

命令： nvidia-smi -l

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

命令：watch -n 3 nvidia-smi

功能：设定刷新时间（秒）显示GPU使用情况

![Center][]

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

在终端执行程序时指定GPU

CUDA\_VISIBLE\_DEVICES=1   python  your\_file.py

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

可用的形式如下：

CUDA\_VISIBLE\_DEVICES=1           Only device 1 will be seen  
CUDA\_VISIBLE\_DEVICES=0,1         Devices 0 and 1 will be visible  
CUDA\_VISIBLE\_DEVICES="0,1"       Same as above, quotation marks are optional  
CUDA\_VISIBLE\_DEVICES=0,2,3       Devices 0, 2, 3 will be visible; device 1 is masked

CUDA\_VISIBLE\_DEVICES=""          No GPU will be visible

在Python代码中指定GPU

import os

os.environ\["CUDA\_DEVICE\_ORDER"\] = "PCI\_BUS\_ID"  
os.environ\["CUDA\_VISIBLE\_DEVICES"\] = "0"

设置定量的GPU使用量

config = tf.ConfigProto()   
config.gpu\_options.per\_process\_gpu\_memory\_fraction = 0.9 \# 占用GPU90%的显存   
session = tf.Session(config=config)

设置最小的GPU使用量

config = tf.ConfigProto()   
config.gpu\_options.allow\_growth = True   
session = tf.Session(config=config)

[Center]: /images/20220514/d51b29615d044ec2a1e8106635fce24a.png

发表评论取消回复

表情：

评论列表（有 0 条评论，261人围观）

还没有评论，来说两句吧...

相关阅读

相关前端薪资，领跑跑跑跑跑跑全国

身为前端工程师，2022年虎年将至，你年初定的小目标实现了多少？改变世界的程序员，引领着时代，见证着时代。今天播妞带大家看看前端程序员的薪资有什么变化！ --------

绝地灬酷狼/ 2023年10月10日 13:39/ 0 赞/ 101 阅读

相关 TensorFlow使用gpu

1.1 关于CUDA： tensorflow-gpu 1.5版本及以上要求CUDA版本为9.0 查看本机CUDA版本方法： cat /usr/local/cuda/v

比眉伴天荒/ 2023年05月30日 06:17/ 0 赞/ 86 阅读

相关 1.使用nsys分析GPU应用程序

例子 GPU 跨步矩阵加法 vector-add.cu include <stdio.h> / Refactor host funct

痛定思痛。/ 2022年12月23日 03:54/ 0 赞/ 339 阅读

相关指定使用哪个GPU运行程序

一：首先使用命令`nvidia-smi`查看有哪些可以使用的GPU 二：指定GPU运行Python程序 CUDA_VISIBLE_DEVICES=1 python

我不是女神ヾ/ 2022年11月21日 06:57/ 0 赞/ 417 阅读

相关【Android】程序跑断点能跑到,正常执行时不执行

程序跑断点能跑到,正常执行时不执行 > 给个思路,执行时间!!! `比如有时候我们退出App,使用exitProcess(0)的时候,很奇怪,它前面的方法,为啥没有执行

àì夳堔傛蜴生んèń/ 2022年10月09日 00:43/ 0 赞/ 271 阅读

相关使用GPU跑程序

查看机器上GPU情况命令： nvidia-smi 功能：显示机器上gpu的情况命令： nvidia-smi -l 功能：定时更新显示机器上gpu的情况命令：wa

绝地灬酷狼/ 2022年05月13日 20:17/ 0 赞/ 262 阅读

相关程序是怎样跑起来的

<table> <tbody> <tr> <td><font> 写在前面</font></td> </tr> </tbody> </table>

清疚/ 2022年05月11日 04:04/ 0 赞/ 326 阅读

相关【tensorflow 学习】 gpu使用

由于tensorflow默认抢占服务器所有GPU显存，只允许一个小内存的程序也会占用所有GPU资源。下面提出使用GPU运行tensorflow的几点建议： 1.在运行之前先查

傷城~/ 2022年04月18日 00:11/ 0 赞/ 817 阅读

相关小程序实现跑马灯效果

跑马灯效果比较常见，一般做电商类的小程序都会使用到; ![在这里插入图片描述][20181112114423109.gif] 跑马灯效果的制作制作方式很简单，先方

素颜马尾好姑娘i/ 2022年04月17日 02:55/ 0 赞/ 705 阅读

相关查看是否用GPU跑的TensorFlow程序

查看是否用GPU跑的TensorFlow程序第一种方法，直接输出日志法（推荐） import tensorflow as tf sess = tf.Session

桃扇骨/ 2022年01月12日 06:11/ 0 赞/ 512 阅读