最全的科学环境配置,CUDA,CUDNN,PyTorch的安装,win10

本文详细介绍了如何检查并升级NVIDIA驱动,选择合适的CUDA和CUDNN版本,以及在已安装Anaconda的基础上安装PyTorch 1.8.0 CPU+GPU的过程,包括驱动卸载、新版本下载、环境配置和版本验证。
摘要由CSDN通过智能技术生成

科学环境的安装

1. 检查自己的NVIDA驱动版本

NVIDA驱动版本也决定了目前可以安装的CUDA的版本的上限,这个也是根据自己想要安装的PyTorch版本来 决定的,下面列出目前最新的一些NVIDA驱动版本、CUDA版本、以及PyTorch-GPU版本的对应情况,自行进行选择。

(1) 首先检查自己的NVIDA驱动版本

image-20210813102728226

通过组件也可查询目前可预装的最高的版本的CUDA。

image-20210813102910705

  • 通过确定PyTorch的预装版本来确定--->预装的CUDA版本确定---->预装的NVIDA版本

image-20210813103426767

PyTorch的版本还可以根据官网previous-version进行自行筛选,这里只提供部分筛选情况。

CUDAToolkit版本可用PyTorch版本
7.50.4.1 ,0.3.0, 0.2.0,0.1.12-0.1.6
8.01.1.0,1.0.0 ,0.4.1
9.01.1.0,1.0.1, 1.0.0,0.4.1
9.21.7.1,1.7.0,1.6.0,1.5.1,1.5.0,1.4.0,1.2.0,0.4.1
10.01.2.0,1.1.0,1.0.1 ,1.0.0
10.11.7.1,1.7.0,1.6.0,1.5.1,1.5.0, 1.4.0,1.3.0
10.21.9.0,1.8.0 ,1.7.1,1.7.0,1.6.0,1.5.1,1.5.0
11.01.7.1,1.7.0
11.11.9.0,1.8.0

(2)下载NVIDA驱动

NVIDA驱动下载,根据自己的笔记本配置需求进行下载(Tips: 不需要下载最新的驱动),傻瓜式安装就可以。

image-20210813104704243

2. 安装CUDA以及CUDNN(这里预装CUDA10.2与CUDNN7.6.5,截止2021.8.10, 10.2版本应该是较稳定的)

(1) 卸载原有版本的CUDA(10.0)以及CUDNN

image-20210813112109576

除了蓝框中的不需要卸载,其他全部卸载。

(2)下载新版本的CUDA10.2以及CUDNN7.6.5

(3)安装、配置环境变量

  • 双击CUDA程序,首先会选择一个安装的临时目录,用来存放首先解压的安装文件,这个默认或者新建一个空的文件夹即可。

  • 使用自定义安装

    image-20210813131949958

  • image-20210813150726568

 

 

如果要安装VIisual Studio Integration,则需要安装 Visual Studio,可以选择安装或者可以取消勾选Visual Studio Integration(这里解释一下,这个模块是对VS编译的支持,没有安装VS无法征常工作,而需要VS辅助则是需要编译cuda程序,这种编译不建议在Windows下进行,一般Windows下能跑深度学习原生框架的代码就行)

在这里如果需要Visual Studio的支持,在这里,以VS2017为例,因为VS的安装势必会带来C盘的大幅度占用,所以这里使用创建软连接的方法。

  • 运行一下vs_community.exe,等弹出安装选择界面后关闭。关闭后,会在C:\ProgramData\Microsoft\VisualStudio 下创建一个Packages,然后,把这个Packages文件夹删掉。

  • 以管理员身份打开命令提示符,运行 mklink /D C:\ProgramData\Microsoft\VisualStudio\Packages D:\vs2017offline命令。D:\vs2017offline 是你下载以及更新后续packages的文件夹的位置。

  • 打开之前安装了一个 Visual Studio Installer(会在你打开vs_community.exe之后安装) ,打开这个软件,就可以安装自己想要的功能了。安装时所有需要的组件都会下载到指定目录,并且不会占用C盘的空间

    image-20210813140859960

  • 将这个三个文件夹放到一起,方便日后查找。

    image-20210813151438199

  • 将CUDNN解压后复制到D:\CUDA_V102\Dvelopment中

    image-20210813152731144

    image-20210813152750332

  • 添加以下环境变量

    image-20210813153301295

  • 检验是否安装完成

    image-20210813153912225

 

 

 

 

3. 安装PyTorch CPU+GPU(在已经安装Anaconda的基础上)

(1)卸载之前的Torch版本

  • pip uninstall torch 
    pip uninstall torchvision

    image-20210813155002418

    image-20210813155631405

 

 

(2)安装Torch==1.8.0

  • 可以通过官网命令查找

    conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

    image-20210813163532978

 

(3) 检查PyTorch-Gpu版本是否可用

>> python
>> import torch 
>> torch.__version__
>> torch.cuda.is_available()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值