【AI】PyTorn安装简述

【AI】PyTorn安装简述

1、PyTorn

• 官网：https://pytorch.org/get-started/locally/

• 直接安装命令：conda install
  pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

• 安装失败：cudatoolkit下载忒慢

2、分别手动下载及安装

（1）选择版本：

PyTorch 版本	CUDA 环境
0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	9.2
1.2.0、1.1.0、1.0.0(1)	10.0
1.4.0、1.5.0(1)、1.6.0、1.7.0(1)	10.1
1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	10.2
1.7.0(1)	11.0
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0	11.1
1.8.0(1)、1.9.0、1.10.0(1/2)、1.11.0、1.12.0(1)	11.3
1.11.0	11.5
1.12.0(1)	11.6

（2）CUDA下载：

①地址：https://developer.nvidia.com/cuda-toolkit-archive
②选择版本(自己)：11.3.1
③安装选项：不要选Visual Studio Integration，不能成功安装；默认的路径
④环境变量安装过程中自动加上了。

（3）cuDNN配置

①地址：https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows
②下载：注册并选择CUDA对应版本
③解压复制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3”

（4）pyTorch安装

①地址：https://pytorch.org/get-started/previous-versions/
②安装：（版本自选对应CUDA，有代码，直接去粘贴）

pip install torch==1.12.0+cu113 torchvision==0.13.0+cu113 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu113 

3、安装完毕，测试一下吧

import torch

print(torch.__version__)
# 1.12.0+cu113

print(torch.cuda.is_available())
# True

print(torch.version.cuda)
# 11.3