单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)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
