驗證 Pytorch 是否安裝成功

一個簡單的測試

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

再來，測試 Pytorch 是否可以使用 GPU 和 CUDA

import torch
torch.cuda.is_available()

結果如下

Reference

1. Pytorch 官網 — Get Started