pytorch入门

一、 pytorch介绍

1.1 什么是pytorch
这是一个基于Python的科学计算软件包，针对两组受众：

NumPy的替代品，可以使用GPU的强大功能
深度学习研究平台，提供最大的灵活性和速度
1.2 为什么选择pytorch
易用性：容易学，易上手，可以快速进行学习与运用
流行性：当前的流行框架之一，另外还有tensorflow等

二、pytorch的基础知识

Pytorch 的安装
因为我一直在使用 Python 工作，所以我使用的是 Python3.6.6版本的
Python，而我的CUDA使用的是CUDA10，所以我直接使用命令安装 Pytorch 就可以了，（如果没有显卡或者没有装CUDA并且不想装CUDA，那么可以使用CPU版本的 Pytorch，具体详细信息可以访问官网查看https://pytorch.org/get-started/locally/）

在这里要注意了，Windows 环境下 Pytorch 只支持Python3.x，不支持Python2.x

pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl # 这个是 Pytorch 的安装包
pip3 install torchvision

验证是否安装成功
命令行输入python
执行下面的代码：
from future import print_function
import torch
x = torch.rand(5,3)
print(x)

输出的结果类似如下就成功了：

tensor([[0.9010, 0.6002, 0.0159],
[0.1889, 0.4830, 0.6653],
[0.1276, 0.7022, 0.6697],
[0.1669, 0.4375, 0.5023],
[0.1307, 0.9552, 0.6480]])

到此安装结束
如果电脑安装了CUDA但是不知道版本是多少，在终端中执行以下代码就可以看到了
V10.0.130 是CUDNN的版本号，前面的10.0就是你的CUDA的版本号