在某些情况下,PyTorch中的GPU的调用是存在一定的循序关系的。
如果想要调整这个顺序,应该使用以下的命令运行PyTorch的训练或者推理程序:
CUDA_VISIBLE_DEVICES=[your_new_gpu_order] python [your_python_file_to_be_run]
比如以下代码会设置第2张卡为“实际运行的第0张卡”:
CUDA_VISIBLE_DEVICES=2,0 python [your_python_file_to_be_run]
在某些情况下,PyTorch中的GPU的调用是存在一定的循序关系的。
如果想要调整这个顺序,应该使用以下的命令运行PyTorch的训练或者推理程序:
CUDA_VISIBLE_DEVICES=[your_new_gpu_order] python [your_python_file_to_be_run]
比如以下代码会设置第2张卡为“实际运行的第0张卡”:
CUDA_VISIBLE_DEVICES=2,0 python [your_python_file_to_be_run]