pytorch 1.11.0 安装流程

• 前言
• 一、CUDA 安装
• 二、8.2.1 cudnn
• 三、安装 pytorch
• 测试

我的是基于 pycharm + Anaconda 安装pytorch

pytorch1.11.0版本需要 11.3.0 CUDA + 8.2.1cudnn

win+R 进入命令 输入

nvidia-smi

查看 Driver Version 是否 大于等于 11.3 CUDA 所需要的版本


CUDA11.3只要大于等于452.39

CUDA各个版本下载点这里

下载11.3.0

这里推荐复制链接地址使用迅雷下载

下载完成后 打开，这只是一个解压路径在哪里不重要

这里你可以改为D盘只需要把默认路径上的C改为D

最后下一步安装完成

cuda对应cudnn

下载CUDA 11.x
下载好cudnn后把里面的文件全部复制粘贴到 CUDA中替换里面的文件

pytorch官网 下载 复制下面的命令

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

如果报以下超时异常
pip3._vendor.urllib3.exceptions.ReadTimeoutError:HTTPSConnectionPool(host=‘download.pytorch.org’, port=443): Read timed out.

跳转这里解决

测试 pytorch 是否安装成功

import torch
print(torch.__version__)
输出的结果
1.11.0+cu113

测试CUDA是否可用

print(torch.cuda.is_available())
输出结果
True

安装成功