PyTorch【1】PyTorch的安装

一.安装适合的CUDA与CUDNN

安装教程 : 安装CUDA和CUDNN

二.安装 torch torchvision torchaudio

进入 PyTorch官方网站 网站，查看符合cuda版本的安装命令。

1.在线安装

如果CUDA的版本为11.2，就在官网查找带有 cu112 关键字符的安装命令，如果没有，可以安装 cu111 或更低的版本，比如如下命令行。

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

2.本地安装（推荐）

由于某些原因，使用方法一安装时下载速度可能极慢，导致安装失败，此时选择本地安装就行。

(1).下载安装文件

进入 官方安装包 网站，直接 Ctrf+F ，输入 cu111 （或其他版本）关键字，在查找结果找到符合系统、符合python版本的两个whl文件:

• torch 的 cu111/torch-xxx.whl
• torchvision 的 cu111/torchvision-xxx.whl

直接右键点击这两个文件使用迅雷下载即可

(2).安装

将下载好的两个文件，复制到新建好的项目文件夹中，在命令行窗口，直接输入下面两个命令即可

• pip install cu111/torch-xxx.whl
• pip install cu111/torchvision-xxx.whl

三.检查是否成功安装

在项目下建立.py文件，输入一下命令，成功显示对应信息即可。

import torch
import torch.nn as nn

print(torch.cuda.is_available())		# True
print(torch.__version__)				# 1.9.1+cu111
print(torch.cuda_version)				# 11.1
print(torch.cuda.current_device())		# 0
print(torch.cuda.device_count())		# 1