浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置
作者:asher_lithium 时间:2021-10-05 06:50:08
服务器有多张显卡,一般是组里共用,分配好显卡和任务就体现公德了。除了在代码中指定使用的 GPU 编号,还可以直接设置可见 GPU 编号,使程序/用户只对部分 GPU 可见。
操作很简单,使用环境变量 CUDA_VISIBLE_DEVICES 即可。
具体来说,如果使用单卡运行 Python 脚本,则可输入
CUDA_VISIBLE_DEVICES=1 python my_script.py
脚本将只使用 GPU1。
在 .py 脚本和 Notebook 中设置,则
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
还可以直接设置临时的环境变量:
export CUDA_VISIBLE_DEVICES="0"
此时该用户的 CUDA 只看得见 GPU0。
至于显存设置,可以设置使用比例(70%):
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
也可以按需增长:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
如果是 Keras 使用 TensorFlow 后端,则可通过如
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))
更改使用设置。
来源:https://blog.csdn.net/leayc/article/details/79707205
标签:GPU,TensorFlow,显存
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
python学习之第三方包安装方法(两种方法)
2021-02-20 03:29:40
![](https://img.aspxhome.com/file/2023/4/99014_0s.png)
实用PHP会员权限控制实现原理分析
2023-11-23 11:32:39
![](https://img.aspxhome.com/file/2023/6/111066_0s.gif)
python数据分析工具之 matplotlib详解
2023-03-11 21:53:24
![](https://img.aspxhome.com/file/2023/0/107040_0s.jpg)
Sql Server 和 Access 操作数据库结构Sql语句
2008-02-11 18:59:00
设计需知:移动网站设计应注意的技术细节
2012-04-20 13:03:23
Windows自动执行python脚本操作步骤
2023-04-22 15:32:42
![](https://img.aspxhome.com/file/2023/1/101731_0s.jpg)
树莓派4B安装Tensorflow的方法步骤
2023-09-05 17:28:06
![](https://img.aspxhome.com/file/2023/7/120267_0s.png)
js检查全角字符正则表达式[\\uFE30-\\uFFA0]
2008-10-30 12:39:00
用Python将结果保存为xlsx的方法
2021-10-22 22:59:34
详解Python中的分组函数groupby和itertools)
2022-08-30 15:03:18
SqlServer表死锁的解决方法分享
2011-09-30 11:16:16
Python 计算机视觉编程进阶之OpenCV 图像锐化及边缘检测
2021-07-31 11:03:29
![](https://img.aspxhome.com/file/2023/8/105308_0s.png)
支持鼠标拖拽的简单目录树代码
2011-07-01 12:34:09
django 发送手机验证码的示例代码
2023-05-07 03:42:37
![](https://img.aspxhome.com/file/2023/7/120377_0s.png)
好友列表利用率越高,社区越失败
2008-04-10 12:22:00
挑战! 纯Javascript 重现经典网游! <魔力宝贝>
2008-10-04 10:37:00
![](https://img.aspxhome.com/file/UploadPic/200810/4/2008104104524850s.jpg)
回调函数的意义以及python实现实例
2021-07-17 11:42:07
pycharm 设置项目的根目录教程
2021-02-05 05:12:31
![](https://img.aspxhome.com/file/2023/0/90450_0s.jpg)
LZ77 算法的JS实现
2010-04-11 22:32:00
从SNS看社会化界面设计 I
2009-02-19 12:21:00
![](https://img.aspxhome.com/file/UploadPic/20092/19/01-43s.jpg)