dvlab-research/PFENet

The gpu setting

Closed this issue · 1 comments

caoql98 commented

I have changed the train_gpu: as 4, why the model still goes on the gpu 0. Could you give me some advice?

tianzhuotao commented

@cql980603 You can try to add 'CUDA_VISIBLE_DEVICES=1' before the training/testing command lines to force the GPU-1 to be used in your server.