单GPU同时跑pytorch和tensorflow

单GPU同时跑pytorch和tensorflow

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