Issue on running on GPU
#1
by
SantiagoTesla
- opened
I somehow can't get it to work on my gpu, it is switching to ram everytime I run it. I tried initializing device map to cuda too, but to no avail. Anyone facing the same issue?