RuntimeError: CUDA sin memoria. Intenté asignar 2,08 GiB (GPU 0 ； 4,00 GiB de capacidad total ； 274,14

RuntimeError: CUDA sin memoria. Intenté asignar 2,08 GiB (GPU 0; 4,00 GiB de capacidad total; 274,14 MiB ya asignados; 2,63 GiB libres; 55,86 MiB en caché)
Solución:
num_works = 4 cambiado a num_works = 0;
batchseiz = 2 cambiado a tamaño de lote = 1;