How to Run the Program on the Second GPU When the First GPU is Occupied?

[xiaosa269]

Hello, thank you very much for your work. My server has two GPUs, but the first one is currently in use by someone else. How can I configure your program to run on the second GPU?

[Gofinge]

# 0 - the first GPU; 1 - the second GPU
export CUDA_VISIBLE_DEVICES=1
sh script/train.sh -g 1 -d XXX -c XXX -n XXX

[SKY-LY777]

为啥显存在第二个GPU上训练，但是现存还是用第一张卡的？

[Gofinge]

环境问题千奇百怪，这个真只能靠自己排查了。通常情况下export CUDA_VISIBLE_DEVICES正确的话，不会出现这个问题。