1. 程式人生 > >tensorflow禁用GPU

tensorflow禁用GPU

如果安裝了tensorflow-gpu版,但是在測試時只想用cpu版本,可以作如下修改:

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"    
import tensorflow as tf

參考連結:https://stackoverflow.com/questions/44552585/prevent-tensorflow-from-accessing-the-gpu