cv2:
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('burano.jpg')
plt.imshow(img)
plt.show()
torch（gpu）:
import torch
import matplotlib.pyplot as plt
x = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y = x.pow(2) + 0.2*torch.rand(x.size())
plt.scatter(x.data.numpy(), y.data.numpy())
plt.show()
DEVICE = torch.device('cuda')
print(torch.version.cuda)
print(torch.backends.cudnn.version())
