Code to use nVidia GPU

[aodor]

I was in the SF training. I have the slides, the environment, te data, but we don't have the code on how to use the examples if we have a GPU. How can I get them?

[pmallari]

To check if CUDA is found, you can print this:

torch.cuda.is_available()

then for training models with cuda, you can move the model and the input data to cuda by doing

model = myModel()
model.cuda()
data    = myData()   #or any feature variable
data.cuda()