关于Pytorch下载并进行部署
目录
1.关于Pytorch
2.Pytorch环境部署前提条件
3.部署方法
4.测试
1.关于Pytorch
PyTorch是一个开源的Python机器学习库，基于Torch，用于计算机视觉、自然语言处理等应用程序。
2017年1月，由Facebook人工智能研究院（FAIR）基于Torch推出了PyTorch。它是一个基于Python的可续计算包，提供两个高级功能：1、具有强大的GPU加速的张量计算（如NumPy）。2、包含自动求导系统的深度神经网络。通过torch，我们可以很便捷地调用封装好的api，避免了自己去封装一个基本的网络类型。
2.Pytorch环境部署前提条件
首先，本次环境部署主要基于Anaconda进行。
其次，Pytorch的部署必须满足以下条件：
- 使用的python环境必须是64位。可以在ananconda propt中通过conda info命令查得：
我们可以从清华镜像网站文件目录中查得，win32平台下是不包含pytorch包的，所以需要使用win64平台进行部署：
- Pytorch的版本需要和python版本对应，否则会无法导入，版本对应关系如下：
3.部署方法
在确保了上述条件后，可以开始部署PyTorch环境。
部署Pytorch环境具体可分为下载cuda包、torch包、torchversion包三个过程。
为了加快Pytorch包下载速度，我们可以配置清华镜像：
在用户文件夹中创建.condarc文件夹，输入以下内容
=
channels:- defaults- conda-forge
show_channel_urls: true
default_channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmsys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudbioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudmenpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudpytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudpytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudsimpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud然后在anaconda创建后的虚拟环境中，输入对应命令，进行下载：
conda install pytorch torchvision torchaudio pytorch-cuda=11.6我们也可以指定对应的版本号：
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2注意命令最后不要夹-c python，否则会使用默认的channels进行下载，而不使用清华镜像。
若还是无法下载，建议直接到 清华镜像网站 寻找对应的torch包，本地下载会比较快：
选择对应的python版本、cuda版本下的torch包：
torchverison包也一样，版本对应参考上文。
cuda包下面文件夹中：
cudnn和cuda版本之间是对应版本的，根据选择好的cuda选择对应的cudnn即可
下载完成后，将压缩包放入操作文件夹中，执行以下命令：
conda install --use-local pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2pytorch-1.6.0-py3.8_cuda102_cudnn7_0.tar.bz2为你下载的压缩包名称
将cuda、cudnn、torch、torchversion四个包安装完成之后，即完成可pytorch环境部署。
4.测试
打开pycharm，输入以下代码：
import torch
print(torch.__version__)输出你的pytorch版本即为部署成功：
