RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 31.75 GiB total capacity; 612.33 MiB already allocated; 14.75 MiB free; 622.00 MiB reserved in total by PyTorch)

[zccoder]

2022-10-24 15:03:50.918 File "/usr/local/python/lib/python3.7/site-packages/torch/nn/functional.py", line 1847, in linear

2022-10-24 15:03:50.918 return torch._C._nn.linear(input, weight, bias)

2022-10-24 15:03:50.918 RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling cublasCreate(handle)