如何在ubuntu中将单个GPU专用于单个笔记本tensorflow 2.2版本

我的系统中有4个GPU,我试图将每个GPU专用于每个笔记本(在我的情况下,将4个GPU分配给4个笔记本)。我发现使用with tf.device('/device:GPU:2')在tensorflow 1中可用,但在Tensorflow 2.2版本中不起作用。我也尝试使用os.environ["CUDA_VISIBLE_DEVICES"]="2"在一个笔记本中访问'/device:GPU:2',但无法实现。您能帮我在Tensorflow 2.2版本中怎么做。

speedisk 回答:如何在ubuntu中将单个GPU专用于单个笔记本tensorflow 2.2版本

为社区的利益在答案部分进行回答。感谢Ghanem的回答。

此问题已通过以下方式解决-

尝试在os.environ [“ CUDA_VISIBLE_DEVICES”] =“ 2”之前添加os.environ [“ CUDA_DEVICE_ORDER”] =“ PCI_BUS_ID”

本文链接:https://www.f2er.com/1322568.html

大家都在问