美文网首页
GPU的具体使用

GPU的具体使用

作者: 黑恶歌王 | 来源:发表于2018-08-29 14:43 被阅读0次
    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的具体使用

          本文链接:https://www.haomeiwen.com/subject/nazhwftx.html