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?

Sign up or log in to comment