Multiple GPU support?

[james-vincent]

We find we must set CUDA_VISIBLE_DEVICES in order to run without error. Setting this to more than one device however we get a file error:

FileNotFoundError: [Errno 2] No such file or directory: 'exp/simsiam2d3d/test_sample/model_20.pth'

Does MiloPYP support more then one GPU?

[huangqinwen]

sorry currently it doesn't support multi-gpu yet. but the code should be able to work with a single gpu.