pytorch基础学习(一) pytorch的安装

1. 下载安装

pytorch是目前应用最宽泛的机器学习库，因其代码框架明晰而广受欢迎。
pytorch的安装也非常简单，进入pytorch官网https://pytorch.org/，可以看到下图：

根据自己的系统版本，安装方式，cuda版本进行选择，直接安装即可。
建议使用conda进行安装，进入anaconda的官网进行下载安装https://www.anaconda.com/.

2. 可能出现的问题

因为pytorch的网站是在国外的，国内安装很可能因为网络问题安装不上，此时可以通过添加清华源或从pytorch库网站https://download.pytorch.org/whl/torch_stable.html下载安装包直接安装（注意先装torch，再装torchvision）。

3.安装完成测试

安装完成后，运行以下语句进行测试：
CPU版:

import torch
import torchvision

t = torch.tensor([1, 2, 3], dtype=torch.float32)
print (type(t))  # <class 'torch.Tensor'>
print (t.dtype)  # torch.float32
print (t.device)  # cpu

output:
<class 'torch.Tensor'>
torch.float32
cpu

GPU版:

import torch
import torchvision

t = torch.tensor([1, 2, 3], dtype=torch.float32).cuda()
print (type(t))  # <class 'torch.Tensor'>
print (t.dtype)  # torch.float32
print (t.device)  # cuda:0

output:
<class 'torch.Tensor'>
torch.float32
cuda:0