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?

Your need to confirm your account before you can post a new comment.

Sign up or log in to comment