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

Pointcept / PointTransformerV3

[CVPR'24 Oral] Official repository of Point Transformer V3 (PTv3)

MIT License

836 stars 47 forks source link

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

Open xiaosa269 opened 2 months ago

xiaosa269 commented 2 months ago

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 commented 2 months ago

# 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 commented 2 days ago

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