1. 程式人生 > >Keras/Tensorflow選擇GPU/CPU運行

Keras/Tensorflow選擇GPU/CPU運行

gpu cuda import rdquo 一個 導入 font env ice

首先,導入os,再按照PCI_BUS_ID順序,從0開始排列GPU,

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" 

然後就可以選擇用哪一個或者那幾個GPU運行:

os.environ[”zCUDA_VISIBLE_DEVICES”] = "0"

用0號GPU,即‘/gpu:0‘運行;

os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1”

用0號和1號設備,即‘/gpu:0‘、‘/gpu:1‘運行;

os.environ[“CUDA_VISIBLE_DEVICES”] = “7”

用7號設備,即‘/gpu:7‘運行,但是‘/gpu:7‘根本不存在,於是就只能用CPU運行了。

Keras/Tensorflow選擇GPU/CPU運行