pycharm用GPU跑代码(tensorflow版) import os import tensorflow as tf config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True)) sess = tf.compat.v1.Session(config=config) # 选择编号为0或1或2的GPU,根据你自己电脑的情况来填 os.environ["CUDA_VISIBLE_DEVICES"] = "0"