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
网友评论