Tensorflow 指定GPU运行
来源:互联网 发布:中国实体经济数据2017 编辑:程序博客网 时间:2024/05/19 16:33
参考网址:http://blog.csdn.net/u014381600/article/details/72911262
1.终端执行程序时设置使用的GPU(linux终端)
CUDA_VISIBLE_DEVICES=1,2 python my_script.py
想指定几块GPU就标明GPU序号即可,中间用逗号隔开
2.python代码中设置使用的GPU(ipython或pycharm调试时)
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
3.设置tensorflow使用的显存大小
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
阅读全文
1 0
- Tensorflow 指定GPU运行
- tensorflow 在指定CPU/GPU上运行方法
- tensorflow 指定GPU
- Tensorflow指定使用哪个GPU
- 指定TensorFlow使用哪一个GPU
- tensorflow指定gpu的使用
- 如何指定GPU运行程序
- Tensorflow 多GPU下,指定使用某一块GPU
- tensorflow中使用指定的GPU及GPU显存
- TensorFlow指定特定GPU以及占用显存的比例
- TensorFlow在训练模型时指定GPU进行训练
- TensorFlow学习系列(一)——GPU指定
- tensorflow-gpu
- 【TensorFlow】Windows环境下PyCharm运行TensorFlow GPU版(附TensorFlow更新方法)
- win10下tensorflow库gpu运行环境配置实操
- Tensorflow+SSD+Windows10+GPU的配置及运行
- (转)tensorflow中使用指定的GPU及GPU显存
- 如何指定程序在某个GPU上运行
- Ubuntu Server 14.04升级Ubuntu Server 16.04
- BZOJ 1375: [Baltic2002]Bicriterial routing 双调路径 SPFA+DP思想
- 实习日志(第七周)
- 精通闭包(自用)
- 安卓圆形头像理解
- Tensorflow 指定GPU运行
- PHP7.1安装yaf扩展
- 【性能测试菜鸟】收集各路大神的各种信息
- db2客户端用dbvisualizer无法进行reorg操作
- maven配置阿里云镜像
- gravity有时会不起作用的问题在于
- JS原型
- LVM
- Python 代码规范