Open juingzhou opened 5 years ago
RuntimeError: CUDA out of memory. Tried to allocate 5.05 GiB (GPU 0; 15.75 GiB total capacity; 10.42 GiB already allocated; 3.68 GiB free; 50.61 MiB cached)
Training on a V100 gpu under the same setting as training demo (--n_resgroups 20 --n_resblocks 10)
Please kindly let me know how to deal with it if this is possible.
RuntimeError: CUDA out of memory. Tried to allocate 5.05 GiB (GPU 0; 15.75 GiB total capacity; 10.42 GiB already allocated; 3.68 GiB free; 50.61 MiB cached)
Training on a V100 gpu under the same setting as training demo (--n_resgroups 20 --n_resblocks 10)
Please kindly let me know how to deal with it if this is possible.