Pytorch_Generalized_3D_Lane_Detection icon indicating copy to clipboard operation
Pytorch_Generalized_3D_Lane_Detection copied to clipboard
verified publisher icon yuliangguo

GPU issue

Open spcrobocar opened this issue 3 years ago • 0 comments

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?

spcrobocar avatar Oct 14 '22 23:10 spcrobocar