在运行别人已经训练好的模型时报了这个错
TensorFlow: CUDA_ERROR_OUT_OF_MEMORY
通过字面意思也知道是内存不够了,而CUDA指的是GPU。而我使用别人的模型,并不希望使用到GPU。为解决这个问题需要进行如下设置:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
在运行别人已经训练好的模型时报了这个错
TensorFlow: CUDA_ERROR_OUT_OF_MEMORY
通过字面意思也知道是内存不够了,而CUDA指的是GPU。而我使用别人的模型,并不希望使用到GPU。为解决这个问题需要进行如下设置:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"