tensorflow下設(shè)置使用某一塊GPU（從0開始編號(hào)）：

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中設(shè)置使用某一塊GPU、多GPU、CPU的操作就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持武林站長(zhǎng)站。