Trouble in training Yolov5
It worked well when the version of pytorch is 1.7, but when I upgrade pytorch to 2.0, alwanys shows ‘CUDA out of memory’. My GPUs are 3090.
And I have added
export CUDA_LAUNCH_BLOCKING=1
export TORCH_USE_CUDA_DSA=1
as the Error suggestions.
Now it shows
RuntimeError: CUDA error: out of memory; Compile with
TORCH_USE_CUDA_DSA
to enable device-side assertions.
What should I do to solve this problem?
torch.cuda.is_available = True