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

2022-03-01 02:26:52

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 0から始まるPCI_BUS_IDの順に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が使用されることを意味します。

最新