• 欢迎访问开心洋葱网站,在线教程,推荐使用最新版火狐浏览器和Chrome浏览器访问本网站,欢迎加入开心洋葱 QQ群
  • 为方便开心洋葱网用户,开心洋葱官网已经开启复制功能!
  • 欢迎访问开心洋葱网站,手机也能访问哦~欢迎加入开心洋葱多维思维学习平台 QQ群
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏开心洋葱吧~~~~~~~~~~~~~!
  • 由于近期流量激增,小站的ECS没能经的起亲们的访问,本站依然没有盈利,如果各位看如果觉着文字不错,还请看官给小站打个赏~~~~~~~~~~~~~!

TensorFlow CPU切换到GPU,CPU/GPU互换

机器学习 开心洋葱 2039次浏览 0个评论

TensorFlow CPU切换到GPUCPU/GPU互换

with tf.Session() as ses:
    with tf.device("/gpu:1"):
        matrix1=tf.constant([[3.,3.]])
        matrix2=tf.constant([[2.],[2.]])
        product=tf.matmul(matrix1,matrix2)

方法二:
<span class="hljs-keyword">import</span> tensorflow <span class="hljs-keyword">as</span> tf sess = tf.Session(config=tf.ConfigProto(device_count={<span class="hljs-string">'gpu'</span>:<span class="hljs-number">0</span>}))

<span class="hljs-keyword">import</span> tensorflow <span class="hljs-keyword">as</span> tf <span class="hljs-keyword">import</span> keras.backend.tensorflow_backend <span class="hljs-keyword">as</span> KTF KTF.set_session(tf.Session(config=tf.ConfigProto(device_count={<span class="hljs-string">'gpu'</span>:<span class="hljs-number">0</span>})))

方法三:使用CUDA_VISIBLE_DEVICES命令行参数,代码如下:

CUDA_VISIBLE_DEVICES="" python3 train.py

Environment Variable Syntax      Results

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
字符 对应的操作
"/cpu:0" The CPU of your machine
"/gpu:0" The GPU of yout machine ,if you have one

等等依次类推:


开心洋葱 , 版权所有丨如未注明 , 均为原创丨未经授权请勿修改 , 转载请注明TensorFlow CPU切换到GPU,CPU/GPU互换
喜欢 (0)

您必须 登录 才能发表评论!

加载中……