OpenNMT Forum

How to use my GPU

I’ve got a

   product: GM204GL [Tesla M60]
   vendor: NVIDIA Corporation

and I want to use it in a PyTorch OpenNMT installation.

I’m running training and getting a ‘Only using CPU message’.

I’ve tried running:

CUDA_VISIBLE_DEVICES=1,3 -world_size 2 -gpu_ranks 0 1

but of course I’ve got no idea what to set for these parameters.

Help!

Also, you can check if CUDA is available in your setup with

import torch
print(torch.cuda.is_available())

or something like that.