文章目录

前言

我的是基于 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 安装

CUDA各个版本下载点这里

下载11.3.0

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

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

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

最后下一步安装完成

二、8.2.1 cudnn

cuda对应cudnn

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

三、安装 pytorch

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

安装成功

来源：太阳是白的