在某些情况下,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]
本文介绍了如何通过设置CUDA_VISIBLE_DEVICES环境变量来调整PyTorch中GPU的运行顺序,以适应不同的计算需求。例如,将第2张GPU设为默认使用的第0张卡的命令是:CUDA_VISIBLE_DEVICES=2,0 python your_python_file_to_be_run,这在多GPU环境的管理和任务调度中非常实用。
在某些情况下,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]
6643
2073
6235

被折叠的 条评论
为什么被折叠?
