单GPU同时跑Pytorch和tensorflow时要先开tensorflow再开pytorch,否则会有cudnn冲突。此外,还需要设置tensorflow占有显存的比例,因为tensorflow默认占用全部显存,显存设置方法:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.8 # 占用80%显存
session = tf.Session(config=config)
单GPU同时跑Pytorch和tensorflow时要先开tensorflow再开pytorch,否则会有cudnn冲突。此外,还需要设置tensorflow占有显存的比例,因为tensorflow默认占用全部显存,显存设置方法:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.8 # 占用80%显存
session = tf.Session(config=config)