tensorflow下設置使用某一塊GPU（從0開始編號）：

import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "1"

多GPU：

num_gpus = 4for i in range(num_gpus):  with tf.device('/gpu:%d',%i):    。。。

只是用cpu的情況

with tf.device("/cpu:0"):

以上這篇在tensorflow中設置使用某一塊GPU、多GPU、CPU的操作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。