您可以使用在笔记本中设置环境变量
os.environ。在初始化TensorFlow以便将TensorFlow限制为第一个GPU之前，请执行以下操作。
import osos.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" # see issue #152os.environ["CUDA_VISIBLE_DEVICES"]="0"
您可以再次检查您是否拥有对TF可见的正确设备
from tensorflow.python.client import device_libprint device_lib.list_local_devices()
我倾向于从实用程序模块（例如notebook_util）使用它
import notebook_utilnotebook_util.pick_gpu_lowest_memory()import tensorflow as tf
