Setting up the device for PyTorch

device_steup.py

#On the tutorial, we use rather use Intel MKL because of its performance and availability to be used on lower end PCs
#Anyway, if you want it, you can try it on CUDA by uncommenting it.

#use_cuda = torch.cuda.is_available() 
use_cuda = False
device = torch.device('cuda' if use_cuda else 'cpu')

FloatTensor = torch.cuda.FloatTensor if use_cuda else torch.FloatTensor
LongTensor = torch.cuda.LongTensor if use_cuda else torch.LongTensor
DoubleTensor = torch.cuda.DoubleTensor if use_cuda else torch.DoubleTensor

torch.manual_seed(0)
np.random.seed(0)
random.seed(0)

if use_cuda:
    torch.cuda.manual_seed(0)
    torch.cuda.manual_seed_all(0)
    torch.backends.cudnn.deterministic = True
    torch.backends.cudnn.benchmark = False