Because a friend of mine managed to fit the unet and train it on GTX1070, but i get this following error:
Runetimeerror : cuda out of memory. Tried to allocate 12mib (GPU 0; 7.77 GB total capacity; 6.12 GB already allocated; 35mb free; 6.22 GB is resvered in total by pytorch)
So kinda confused. This happens when it goes inside the training phase.