Invalid argument on docker load (GPU-3.1)

I installed today a new GPU - Nvidia RTX 2060,
moved to the GPU-3.1 package and it works very fast - less than 0.5 second (compared to 3.6 seconds with CPU).

After restart I get the following error:
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=663 error=11 : invalid argument

it seems that it still works.

There’s a post about the same problem with RTX cards and CUDA: