我的系统中有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版本中怎么做。
如何在ubuntu中将单个GPU专用于单个笔记本tensorflow 2.2版本
•
问答
speedisk 回答:如何在ubuntu中将单个GPU专用于单个笔记本tensorflow 2.2版本
为社区的利益在答案部分进行回答。感谢Ghanem的回答。
此问题已通过以下方式解决-
尝试在os.environ [“ CUDA_VISIBLE_DEVICES”] =“ 2”之前添加os.environ [“ CUDA_DEVICE_ORDER”] =“ PCI_BUS_ID”