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)
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)