说实话老子百度了各种方法,根本就不行,为啥?
都是一两行代码的回答,和没回答是一样的。这特么要是能用我会写这个博文
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
或者这种:有个毛区别
CUDA_VISIBLE_DEVICES=0,1 python3 train.py
都不能解决问题,统统都是垃圾。
还有一种tf.ConfigProto,这玩意我之前去掉了,这是另外的那个模型的。下面我试试放到TTS的模型中,看看有没有屁用。
事实证明这种也是狗屎一样垃圾。
session_conf = tf.ConfigProto(
device_count={'CPU': 3, 'GPU': 3},
gpu_options=tf.GPUOptions(allow_growth=True),
log_device_placement=False
)