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.


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

import torch

or something like that.