python下使用GPU运行代码(2020更新)

python下使用GPU运行代码(2020更新)

下面展示核心代码,只需调用os包即可实现,CUDA_VISIBLE_DEVICES指GPU块,根据自己的书写即可。我看了其他博客是并没有".v1."这一个代码,去掉v1对应tensorflow1.x,而如果你是tensorflow2.x,则去掉v1会报错,爬了好多坑的心得。

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.compat.v1.Session(config=config)

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