如果你是用的GPU版本的Tensorflow,你可以这样来使用CPU版本的Tensorlfow:
config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)
黄世宇/Shiyu Huang's Personal Page:https://huangshiyu13.github.io/
在CPU上运行Tensorflow
阅读 61
2022-07-18
如果你是用的GPU版本的Tensorflow,你可以这样来使用CPU版本的Tensorlfow:
config = tf.ConfigProto(
device_count = {'GPU': 0}
)
sess = tf.Session(config=config)
黄世宇/Shiyu Huang's Personal Page:https://huangshiyu13.github.io/
相关推荐
精彩评论(0)