os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0"

os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID” # 按照PCI_BUS_ID順序從0開始排列GPU設備 
os.environ[“CUDA_VISIBLE_DEVICES”] = “0” #設置當前使用的GPU設備僅為0號設備  設備名稱為'/gpu:0'
os.environ[“CUDA_VISIBLE_DEVICES”] = “1” #設置當前使用的GPU設備僅為1號設備  設備名稱為'/gpu:0'
os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1” #設置當前使用的GPU設備為0,1號兩個設備,名稱依次為'/gpu:0'、'/gpu:1'
os.environ[“CUDA_VISIBLE_DEVICES”] = “1,0” #設置當前使用的GPU設備為1,0號兩個設備,名稱依次為'/gpu:0'、'/gpu:1'。表示優先使用1號設備,然后使用0號設備
---------------------
作者：James-J
來源：CSDN
原文：https://blog.csdn.net/Jamesjjjjj/article/details/83414680
版權聲明：本文為博主原創文章，轉載請附上博文鏈接！