Pytorch

# uname -a
Linux yundream 5.0.0-40-generic #44-Ubuntu SMP Wed Jan 15 02:03:45 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux
# cat /etc/issue
Ubuntu 19.04 \n \l

# pip --version
pip 18.1 from /home/yundream/virtualenvironment/pytorch/lib/python3.7/site-packages/pip (python 3.7)
# python --version
Python 3.7.3

# nvidia-smi       
Sun Mar  1 23:47:21 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.14       Driver Version: 430.14       CUDA Version: 10.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 105...  Off  | 00000000:06:00.0  On |                  N/A |
| 40%   32C    P5    N/A /  75W |    604MiB /  4036MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

# pip install torch torchvision

from __future__ import print_function
import torch

x = torch.rand(5,3)
print(x)

tensor([[0.5268, 0.3673, 0.5628],
        [0.6845, 0.9954, 0.1437],
        [0.3880, 0.6720, 0.7744],
        [0.0640, 0.4153, 0.3090],
        [0.0252, 0.8269, 0.5751]])

import torch
torch.cuda.is_available()