single GPU

[yunqi831017]

Hello, may I ask how to set up if I want to train with a single GPU

[hello-wwy]

You could set multi_gpu to False in train_settings/tomp/tomp_v1.py and specify gpu through CUDA_VISIBLE_DEVICES. Then, you can train with a single gpu by running run_training.py.

[wwwshengbo]

You could set multi_gpu to False in train_settings/tomp/tomp_v1.py and specify gpu through CUDA_VISIBLE_DEVICES. Then, you can train with a single gpu by running run_training.py.

这个 set multi_gpu to False，训练时还是没有用到 我仅有的一张 GPU，这是什么情况？检查一下，发现这个可以解决https://blog.csdn.net/qq_46600553/article/details/118767360

[goodnight111111111]

您好，非常感谢您的贡献，可以分享一下您的可视化代码