tensorflow gpu和cpu代码区别
发布网友
发布时间：2022-04-29 10:56
共1个回答
热心网友
时间：2022-06-26 13:15
下载时：
pip install tensorflow-gpu
使用时：
需要设置GPU，即需要加上
import tensorflow as tf
gpus = tf.config.list_physical_devices("GPU")
if gpus:
tf.config.experimental.set_memory_growth(gpus[0], True) #设置GPU显存用量按需使用
tf.config.set_visible_devices([gpus[0]],"GPU")
