windows下pytorch(gpu)如何安装,看这就够了——史上最全


文章目录

  • windows下pytorch(gpu)如何安装,看这就够了——史上最全
  • 前言
  • 一、安装anaconda。
  • 二、安装cuda和cudnn
  • 三、安装pytorch



前言

教你如何安装anaconda、cuda、cudnn,pytorch以及它们对应的版本关系。


一、安装anaconda。

anaconda安装后就不用安装python,anaconda就是python的一个集合包,它把很多python必须的依赖包都集成在一起了。

进入anaconda官网,我们可以选则下载最新版的anaconda。如需要历史版本的anaconda可以进入清华anaconda镜像源https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,进行下载。

pytorch一般安装多久 pytorch必须要装anaconda吗_机器学习


anaconda版本并不是越新越好,推荐下载anaconda3.5左右的版本,也就是3.5.0、3.5.1、3.5.2这些版本均可。下载完毕后双击安装包,

pytorch一般安装多久 pytorch必须要装anaconda吗_python_02


点击next

pytorch一般安装多久 pytorch必须要装anaconda吗_机器学习_03


同意使用说明

pytorch一般安装多久 pytorch必须要装anaconda吗_python_04

这儿选择all users(意思是每个电脑账户都可以使用anaconda)

pytorch一般安装多久 pytorch必须要装anaconda吗_深度学习_05


一般安装在d盘

pytorch一般安装多久 pytorch必须要装anaconda吗_pytorch一般安装多久_06


建议都勾选,勾选第一个可以省略把anaconda写入环境变量的操作,很是省事,等待安装完成即可。

二、安装cuda和cudnn

要安装cuda必须要有nvidia型号的显卡,广义的说,九代、十代和以后的显卡都支持。其实之前的显卡不少也是支持的,但其性能不是太好,并不建议安装gpu版本的pytorch。

下载地址(官网):

cuda下载地址https://developer.nvidia.com/cuda-toolkit-archive

pytorch一般安装多久 pytorch必须要装anaconda吗_pytorch一般安装多久_07

cudnn下载地址https://developer.nvidia.com/rdp/cudnn-archive

pytorch一般安装多久 pytorch必须要装anaconda吗_机器学习_08

建议下载cuda10以后的版本,cudnn页面有对应的cuda型号,选择相应的即可。
注意:官网下载要注册nvidia账号,其次显卡驱动要求比较新,建议现在nvidia官网把自己显卡对应的驱动进行升级操作。

双击安装cuda,一定要安装在c盘。

pytorch一般安装多久 pytorch必须要装anaconda吗_pytorch一般安装多久_09

选择自定义选项,一直继续即可。

pytorch一般安装多久 pytorch必须要装anaconda吗_pytorch_10

把下载的cudnn压缩包解压,然后把这四个文件全部复制粘贴到cuda安装的文件夹下

pytorch一般安装多久 pytorch必须要装anaconda吗_pytorch_11

pytorch一般安装多久 pytorch必须要装anaconda吗_深度学习_12

复制粘贴到这儿,同名的全部替换。

至此cuda和cudnn安装完毕。

三、安装pytorch

建议把pytorch的whl文件下载后进行本地安装,注意下载cuda,torch,torchvision对应相匹配的版本。

pytorch和torchvision下载地址https://download.pytorch.org/whl/torch_stable.html

打开anaconda prompt
创建一个使用pytorch的新环境

conda create -n pytorch-gpu python=3.7

进入创建的新环境

activate pytorch-gpu

进入pytorch和torchvision下载好的文件目录,进行安装

pip install torchvision-0.8.1+cu101-cp37-cp37m-win_amd64.whl

安装完成后,测试是否安装成功,输入

python
import torch

没有报错,即说明安装成功。