[解决方法]Tensorflow强制使用CPU
·
在导入Keras/TensorFlow之前：
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"] = ""
如果上述命令不可用，可以尝试下述语句：
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。
更多推荐
1
0- 0
已为社区贡献6条内容
所有评论(0)