When I run train_net, error occurs like:
return F.conv2d(input, weight, self.bias, self.stride,
RuntimeError: CUDA out of memory. Tried to allocate 12.00 MiB (GPU 0; 7.79 GiB total capacity; 6.48 GiB already allocated; 11.12 MiB free; 6.52 GiB reserved in total by PyTorch)
I was wondering whether the error is the one that my GPU is not supported by batch_size=64 as your mentioned. My GPU is 2080 with 16G memory. How can I fix it, is any configuration instruction I can use?
Thanx for your help.
When I run train_net, error occurs like:
return F.conv2d(input, weight, self.bias, self.stride,
RuntimeError: CUDA out of memory. Tried to allocate 12.00 MiB (GPU 0; 7.79 GiB total capacity; 6.48 GiB already allocated; 11.12 MiB free; 6.52 GiB reserved in total by PyTorch)
I was wondering whether the error is the one that my GPU is not supported by batch_size=64 as your mentioned. My GPU is 2080 with 16G memory. How can I fix it, is any configuration instruction I can use?
Thanx for your help.