在机器学习中,如果不设置GPU使用情况,会默认使用第一块GPU,而且会完全占用所有GPU资源。使用下面设置就能轻松解决问题。 1 2 3 $ import os $ o s.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" $ os.environ["CUDA_VISIBLE_DEVICES"]="0"