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/,进行下载。
anaconda版本并不是越新越好,推荐下载anaconda3.5左右的版本,也就是3.5.0、3.5.1、3.5.2这些版本均可。下载完毕后双击安装包,
点击next
同意使用说明
这儿选择all users(意思是每个电脑账户都可以使用anaconda)
一般安装在d盘
建议都勾选,勾选第一个可以省略把anaconda写入环境变量的操作,很是省事,等待安装完成即可。
二、安装cuda和cudnn
要安装cuda必须要有nvidia型号的显卡,广义的说,九代、十代和以后的显卡都支持。其实之前的显卡不少也是支持的,但其性能不是太好,并不建议安装gpu版本的pytorch。
下载地址(官网):
cuda下载地址https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载地址https://developer.nvidia.com/rdp/cudnn-archive
建议下载cuda10以后的版本,cudnn页面有对应的cuda型号,选择相应的即可。
注意:官网下载要注册nvidia账号,其次显卡驱动要求比较新,建议现在nvidia官网把自己显卡对应的驱动进行升级操作。
双击安装cuda,一定要安装在c盘。
选择自定义选项,一直继续即可。
把下载的cudnn压缩包解压,然后把这四个文件全部复制粘贴到cuda安装的文件夹下
复制粘贴到这儿,同名的全部替换。
至此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
没有报错,即说明安装成功。