Pytorch深度学习（一）Pytorch安装

1 Pytorch介绍

Pytorch是由Torch7团队开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。 能够在强大是GPU加速基础上实现张量和动态神经网络。

1.1 Pytorch提供了两种高层面的功能

(1) 使用强大的GPU加速的Tensor计算（类似numpy）

(2) 构建于基于autograd系统的深度神经网络

1.2 使用Pytorch的原因

（1）作为numpy的替代，以便使用强大的GPU加速；

（2）将其作为一个能提供最大灵活性和速度的深度学习研究平台

2 安装Pytorch

​ Pytorch的安装非常方便，可以使用Anaconda进行安装，也可以使用pip进行安装。在pytorch官网上https://pytorch.org/选择与自己系统环境相匹配的pytorch。

CPU安装：

conda安装

conda install pytorch torchvision torchaudio cpuonly -c pytorch

pip安装

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

GPU安装：

（1）安装NVIDIA驱动

下载地址：https://www.nvidia.cn/Download/index.aspx?lang=cn
（2）安装CUDA

安装CUDA Driver时，版本需与NVIDIA GPU Driver的版本一致，这样CUDA才能找到显卡。

（3）安装cuDNN

cuDNN是用于深度神经网络的GPU加速库。注册NVIDIA并下载cuDNN包，获取地址为https://developer.nvidia.com/rdp/cudnn-archive

（4）安装Python及Pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge