gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
上面这两个表示是对gpu设置操作，之前都要先import tensorflow as tf，这个是基本操作。这里的设置实际上相当是给每块gpu只占用33%的空间，比如12g的大概只占4g左右。
然后还有一点就是如果想指定多块gpu的话，可以用这个比较简单的方法。
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'
这样的话直接就指定了4块gpu，再加上之前写的代码，就可以保证最后的使用。
再提一点就是tf.ConfigProto这个函数就是用来搞gpu的。
上面这两个表示是对gpu设置操作，之前都要先import tensorflow as tf，这个是基本操作。这里的...
$ nvidia-smi 可以查询具体GPU的使用情况
为什么要使用GPU编程 OpenGL 使用的的是GPU的高并发能力，那为什么要使用GPU而不是CPU了？ CPU是...
查看GPU信息，不是使用nvidia-smi，而是使用tegrastats 让docker使用gpu，使用下面的代...
一、深度学习的部署流程 步骤1:确定部署环境ü有没有GPU?具体GPU型号是什么?ü具体什么操作系统?可以联网...
最近被安排研究ncnn的效果，ncnn不能支持gpu加速。 先说一下具体环境，使用的是Ubuntu16.04 首先...
下面介绍一下具体的优化方式，按照硬件划分分为CPU，GPU，内存&硬盘。 CPU Top10 使用Profile找...
使用GPU运行TensorFlow System Information OS: Windows 10 GPU: ...
python代码中设置使用的GPU 如果要在python代码中设置使用的GPU（如使用pycharm进行调试时），...
查看gpu使用情况，发现gpu显存（memory-usage）满了，但是使用率（gpu-util）是0。上面的进程...
本文标题：GPU的具体使用
本文链接：https://www.haomeiwen.com/subject/nazhwftx.html
网友评论