win10+Anaconda+pytorch+CUDA11.1 详细安装指南

一、电脑配置检查

1. 检查电脑显卡类型

注意电脑显卡不是NVIDIA的忽略这一步,非NVIDIA显卡不能安装CUDA。
在桌面鼠标右键:

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第1张图片

点击NVIDIA 控制面板->选择左下角 系统信息->组件

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第2张图片win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第3张图片win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第4张图片

从第三幅图可以看出我的CUDA是11的版本,所以下面我将安装11版本的CUDA

二、更新你的NVIDIA驱动

进入NVIDIA官网:https://www.nvidia.cn/geforce/drivers/  下载GEFORCE EXPERIENCE 方便更新驱动(自己也可以手动跟新,网上有许多教程,我用的是GEFORCE EXPERIENCE 来更新的驱动的)

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第5张图片

下载完之后进行安装,再登陆应用,里面有提示是否跟新驱动,如果需要一定跟新;更新完驱动之后就可以进行cuda以及cudnn的安装了

三、安装CUDA

1.NVIDIA开发者网站下载CUDA 10.1

https://developer.nvidia.com/cuda-downloads 根据自己的版本选择后下载

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第6张图片

直接从官网下载很慢,有好心人已经把安装包放到了百度云网盘,直接从(cuda11+cudnn11百度网盘下载 https://pan.baidu.com/s/1IV_lBCeFFM712xx_iXnhqQ 提取码:0pr5 )下载cuda以及cudnn

2.安装CUDA 10.1

双击运行

设置用于提取CUDA安装程序的临时目录

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第7张图片

提取完成后,经过一段漫长的兼容性检查

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第8张图片

选择自定义安装

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第9张图片

自定义安装选项的时候,CUDA是核心组件必须安装,如果NVIDIA GeForce Experience等组件的当前版本比新版本低可以选择安装

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第10张图片

选择安装位置一般不要更改默认位置

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第11张图片

提示没有Visual Studio,可以不用理会,勾选并点击NEXT。(注意关闭安全软件,否则会失败)

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第12张图片

安装完毕,结束

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第13张图片

4.验证CUDA安装

打开命令行工具(win+r:运行 cmd)

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第14张图片

 

输入nvcc -V (查看版本信息),安装成功

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第15张图片

四、安装cuDNN

1.下载cuDNN

官网: https://developer.nvidia.com/cudnn 下载需要登陆账号,登陆界面“Join”可以注册账号,若不想注册可以将下载链接复制到迅雷等其他下载器中下载

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第16张图片

官网下载很麻烦也很慢,在上面网盘里已经提供了cudnn安装包,下面直接讲解如何安装

2.安装cuDNN

解压下载的压缩文件,压缩出以下几个文件

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第17张图片

将bin、include、lib 粘贴复制到下面的文件夹,直接覆盖掉

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第18张图片

3.cuDNN的环境变量配置

此电脑->属性->高级环境设置->环境变量

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第19张图片

打开之后如下图所示,由于我已经配置好了,图中没有的部分,就是你需要增加的变量;

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第20张图片

在系统中添加以下 5个 环境变量:
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1
(这是之前安装cuda是默认安装位置的路径,如果自定义路径后,需要修改成对应的自定义路径)
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64
CUDA_BIN_PATH = %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

添加方法:按照名称和路径添加即可。

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第21张图片

上面添加完之后我们添加path,双击Path

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第22张图片

添加以下变量:


C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1\common\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1\bin\win64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第23张图片

这样cudnn配置完成;

五、安装Anaconda3 (很容易安装,这里不再一一说明)

官网:https://www.anaconda.com/ 自行下载

六、创建管理虚拟环境

安装好Anaconda3且配置完之后,就可以配置pytorch所需要的虚拟环境了;

打开命令行工具点击Anaconda Prompt

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第24张图片

创建一个名为PyTorchLearn的虚拟环境,选择python 3.6版本(3.6版本的python比较稳定)
在刚才打开的窗口输入:conda create -n PyTorchLearn python=3.6
然后激活PyTorchLearn虚拟环境
输入口令:conda activate PyTorchLearn
最后i退出虚拟环境
输入口令:conda deactivate

 

七、安装PyTorch CUDA版

1.添加清华源

打开命令行工具点击Anaconda Prompt

输入口令: activate PyTorchLearn  

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第25张图片我的虚拟环境名称是pythorch_(当初打错了,现在没改),你们按上面来的话,环境名称是PyTorchLearn  

将下面三个命令依次输入进去
  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

  3. conda config --set show_channel_urls yes

 

2.官网获取安装命令

官网:https://pytorch.org/ 选择对应版本,复制安装命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第26张图片

 

 

 

然后输入pytorch口令:conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

 

你按回车键之后,等一会它会加载出将要安装的一些安装包名称以及大小(由于我已经安装好了,就加载不出来了),但是不影响你的安装,按照下面的步骤来;

其主要三个大包是pytorch、torchvision、cudatoolkit,如果直接安装,几乎是失败的,所以我们需要先在清华源上 ,下载出这三个安装包,如下图所示;

我们从下面地址中下载这三个大包到桌面文件夹(我在桌面新建了一个名为q的文件夹,将下面下载的安装包都放进去了):

cudatoolkit安装包地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第27张图片

Pytorch安装包地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第28张图片win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第29张图片

最终下载了这三个文件:

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第30张图片

在Anaconda Prompt的虚拟环境中我们输入口令:

conda install C:\Users\49660\Desktop\q\cudatoolkit-11.0.221-h74a9793_0.tar.bz2  

口令是由conda install  +  你安装包的对象名称(右击你安装包,点击属性,属性里面的安全中 就可以看见对象名称),如下图所示

win10+Anaconda+pytorch+CUDA11.1 详细安装指南_第31张图片

将这个三个安装包按上面的方法都安装好,最后输入pytorch口令,安装剩下的小型安装包:conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

这样一个完整GPU的pytorch 安装完了

 

3.验证PyTorch安装

在PyTorchLearn虚拟环境中,依次运行下列命令/程序:

python
import torch
torch.cuda.is_available()

运行结果如下;

显示为出True,即安装成功!!!

参考:

1.https://blog.csdn.net/watermelon1123/article/details/88122020?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-4.control

2.https://blog.csdn.net/CZZ_CS/article/details/106541048ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522161128971316780269878122%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25255Fall.%252522%25257D&request_id=161128971316780269878122&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_v2~rank_v29-5-106541048.pc_search_result_before_js&utm_term=Anconada+pytorch

3.https://blog.csdn.net/weixin_48994268/article/details/110392149?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522161129438416780262550853%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25255Fall.%252522%25257D&request_id=161129438416780262550853&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_v2~rank_v29-3-110392149.pc_search_result_before_js&utm_term=cudnn%E7%9A%84%E7%8E%AF%E5%A2%83%E5%8F%98%E9%87%8F%E9%85%8D%E7%BD%AE

 

你可能感兴趣的:(win10+Anaconda+pytorch+CUDA11.1 详细安装指南)