Tensorflow は実行する CPU または GPU を選択します

コード

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_EDVICES"] = '1' #选择01的话是GPU-1的话CPU

おすすめ

転載: blog.csdn.net/qq_43802454/article/details/121219493