转自 https://www.cnblogs.com/excellent-ship/p/8604941.html
import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
注:上述代码一定要放在import tensorflow或keras等之前,否则不起作用。
本文介绍了一种方法,可以在不使用GPU的情况下运行TensorFlow,这对于没有GPU资源或希望节省GPU资源的场景非常有用。通过设置环境变量,可以确保TensorFlow只在CPU上运行。
转自 https://www.cnblogs.com/excellent-ship/p/8604941.html
import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
注:上述代码一定要放在import tensorflow或keras等之前,否则不起作用。
您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
4803
448

被折叠的 条评论
为什么被折叠?