使用CUDA_VISIBLE_DEVICES,我可以在python文件中隐藏设备,但是我不确定如何在jupyter notebook中执行此操作。 那么,能否将某个GPU设置为在同一服务器上运行的某个notebook不可见?或者说设置GPU对其他notebook隐藏? 最佳回答 您可以使用os.environ在笔记本中设置环境变量。在初始化TensorFlow以便将TensorFlow限制为第一个GPU...
您可以使用os.environ在笔记本中设置环境变量。在初始化 TensorFlow 之前执行以下操作以将 TensorFlow 限制为第一个 GPU。 importos os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"# see issue #152os.environ["CUDA_VISIBLE_DEVICES"]="0" 您可以仔细检查您是否拥有对 TF 可见的正确设备 fromtensorflow.python....
CUDA_DEVICE_ORDER=PCI_BUS_IDenvCUDA_VISIBLE_DEVICES=0 请注意,所有环境变量都是字符串,因此无需使用"。您可以通过运行以下命令验证环境变量是否已设置:%env <name_of_var>。或者用%env检查所有这些。