Pytorch简介

pyTorch是一个Python的开源机器学习库。它用于自然语言处理等应用程序。它最初由Facebook人工智能研究小组开发，而优步的Pyro软件则用于概率编程。最初，PyTorch由Hugh Perkins开发，作为基于Torch框架的LusJIT的Python包装器。有两种PyTorch变种。
PyTorch在Python中重新设计和实现Torch，同时为后端代码共享相同的核心C库。PyTorch开发人员调整了这个后端代码，以便有效地运行Python。他们还保留了基于GPU的硬件加速以及基于Lua的Torch的可扩展性功能。

Pytorch[GPU版本]环境部署

因在线pip和anaconda安装问题层出不穷(也可能是因本人能力原因)，本文采用离线安装的方式进行讲解。
环境：pycharm、anaconda[python3.9版本]、windows
本文为采用创建虚拟环境的方式进行操作，而是直接在anaconda环境下安装torch
离线whl网站：https://download.pytorch.org/whl/

推荐系统用pytorch还是 pytorch用什么软件_pytorch

1、cudnn

  1. 首先cmd命令行窗口，输入
nvidia-smi

推荐系统用pytorch还是 pytorch用什么软件_pytorch_02


本处我电脑是CUDA版本11.1，下载安装的torch匹配的cuda版本应在电脑本机CUDA版本及其以下。（本处我电脑安装cuda版本最高支持到11.1）

CUDA官网下载链接，下载速度可能略慢

CUDA下载官网 本处版本号参考前两位即可，第三位为小更新，不影响总体安装环境配置。

推荐系统用pytorch还是 pytorch用什么软件_推荐系统用pytorch还是_03


2. 下载完时候双击打开，进行安装， 记住不要勾选visual code

推荐系统用pytorch还是 pytorch用什么软件_CUDA_04

  1. 如果你是第一次安装，尽量全选 如果你是第n次安装，尽量只选择第一个，不然会出现错误。

推荐系统用pytorch还是 pytorch用什么软件_pytorch_05

  1. 位置不懂的话，默认即可，如果改动，需要记住位置，本人直接默认了。

推荐系统用pytorch还是 pytorch用什么软件_CUDA_06


到此就安装完成了，如果测试自己安装是否成功，在cmd中输入

nvcc -V

推荐系统用pytorch还是 pytorch用什么软件_人工智能_07


显示版本号成功！

2、torch

本次采取是离线安装Torch，需要对应好torch和torchvision版本 以及操作系统（windows64、32 、Linux等等），
安装之前先去网站查询一下，以防没有对应的离线安装包离线安装包网站

1首先点击torch，找到你的cuda版本和操作系统版本对应的torch版本，以及python版本。（此处对应错误，后面无法正常启用GPU）

推荐系统用pytorch还是 pytorch用什么软件_pytorch_08


cu是你下载的cuda版本，101即为10.1，cp是python版本35就是3.5，win_arm64就是64位windows系统，

torch之后的1.5.1就是你下载的torch版本，

3、torchvison

同理，根据你下载的进入torchvision下载对应的版本，torch和torchvison缺一不可。

推荐系统用pytorch还是 pytorch用什么软件_推荐系统用pytorch还是_09

安装

  1. 进入anaconda prompt
  2. cd进入需要安装的python目录下

推荐系统用pytorch还是 pytorch用什么软件_python_10

  1. 使用 pip install 拖入路径 回车安装即可，安装torch和torchvision，安装两个，本操作不做演示。

验证是否安装成功代码：

import torch
 
print(torch.__version__) # 这个输出torch版本
 
print(torch.cuda.is_available()) # 这个输出结果必须为True