1. 程式人生 > >tensorflow 在py中設定GPU的使用

tensorflow 在py中設定GPU的使用

1.  在py檔案中加入

   import os

  os.environ['CUDA_VISIBLE_DEVICES'] = '-1' //-1代表使用cpu,0 1 2 3代表使用哪個GPU,'0,1'代表使用GPO 0和1

2.在py檔案中加入

config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)