RuntimeError: CUDA error: out of memory; Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions

vision
1

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

2

Could you post a script (e.g., with random/dummy data) that reproduces the issue/high memory usage that worked in 1.7 but crashes in 2.0?

4

I’m sorry for bothering you. The issue disappeared on the second day, so I forgot my question. It may because of the unkilled processes, and coincidently I upgraded pytorch, so I ascribe the error to pytorch…