search

yuliangguo / Pytorch_Generalized_3D_Lane_Detection

[ECCV 2020] Official PyTorch implementation of 'Gen-LaneNet: a generalized and scalable approach for 3D lane detection'
307 stars 78 forks source link

GPU issue #26

Open spcrobocar opened 2 years ago

spcrobocar commented 2 years ago

I have two GPU on my desktop, however, even I set os.environ["CUDA_VISIBLE_DEVICES"] = "1", the main_train_GenLaneNet_ext program sill uses the gpu 0. How can I make it use gpu 1?