文章目录
前言
我的是基于 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
最后下一步安装完成
; 二、8.2.1 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
安装成功
Original: https://blog.csdn.net/A8186/article/details/125280251
Author: 太阳是白的
Title: pytorch 1.11.0 安装流程
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/628390/
转载文章受原作者版权保护。转载请注明原作者出处!