Open cravisjan97 opened 3 years ago
When I try to run main.py on two or more GPUs I get RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on the device: cuda:1. How do I solve this issue as my single GPU has low memory
main.py
RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on the device: cuda:1
When I try to run
main.pyon two or more GPUs I getRuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on the device: cuda:1. How do I solve this issue as my single GPU has low memory