pycharm用GPU跑代码(tensorflow版)

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"

你可能感兴趣的:(tensorflow,pycharm,深度学习)