pytorch使用gpu

import cv2

import dlib

import torch as t

if name == ‘main’:
t.cuda.current_device()
t.cuda._initialized = True

x = t.rand(5, 3)

y = t.rand(5, 3)

if t.cuda.is_available():
    x = x.cuda()

    y = y.cuda()

    print(x + y)