博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
第二篇:win10下CUDA+cuDNN+pytorch的安装配置
阅读量:3978 次
发布时间:2019-05-24

本文共 1234 字,大约阅读时间需要 4 分钟。

win10下cuda+cudnn+pytorch的安装配置

其实网上已经有很多关于这方面的博文,自己也已经安装过无数次了,所以这里也来详细记录一下完整环境配置过程,希望可以帮助到刚接触到这方面的新手小白,每一步我都会附上截图的,所以不要怕不会操作哦。


文章目录


一、首先查看电脑可适配的cuda版本

第一种:win+R——输入cmd打开dos界面,输入nvidia-smi

在这里插入图片描述
第二种:利用win10左下角的搜索功能NVIDIA Control Panel打开后——帮助——系统信息——组件:
在这里插入图片描述
然后可以看到我的笔记本支持的是11.2版本的cuda,cuda版本是向下兼容的,这里查询到的是电脑支持的最高的cuda版本。


二、cuda的下载安装

1.官网下载

链接:

在这里插入图片描述
按照下面的步骤依次选择:
在这里插入图片描述
下载之后的文件,双击它进行安装:
在这里插入图片描述

2. CUDA安装

因为我已经在电脑上安装过啦,所以有些截图不是自己的,就在网上找的,除了版本与我自己的不一致其它的没有影响。

2.1 第一个界面:这个路径默认就好了,只是个临时解压目录,安装结束后,会自动删除的,不用管的,点击ok。

在这里插入图片描述
2.2 第二个界面:要经历小小的等待,
在这里插入图片描述
在这里插入图片描述
第三个界面:选择自定义安装界面
在这里插入图片描述
在这里插入图片描述
第四个界面:选择安装位置,这里不要做任何的修改,默认就好啦,然后下一步。
在这里插入图片描述
等待一段时间,点击关闭就好啦!
在这里插入图片描述
然后我们去终端测试一下,使用nvcc -V或者nvcc --version两个命令都可以:
在这里插入图片描述


三、环境变量的配置

先查看一下,cuda安装完成后下面会默认有的:

在这里插入图片描述
进入到环境变量的编辑:
在这里插入图片描述


四、cuDNN的下载安装

1. 官网下载

链接:

需要注意的是cuDNN下载可能需要登录NVIDIA,没有的话需要创建账号哦!

在这里插入图片描述

2. cuDNN安装

下过来就是这样一个文件夹:

在这里插入图片描述
打开是这样的,有三个文件夹:
在这里插入图片描述
然后呢将上面cuDNN中的三个文件夹分别copy到cuda文件夹中就行啦。
在这里插入图片描述
到这里cuda以及cuDNN的安装配置就大功告成啦!


五、pytorch的下载安装

1. 官网下载

链接:

注意这里的箭头所指的一些选项,然后复制使用Run this Command所给的指令,在dos中直接运行。

在这里插入图片描述

2.pytorch安装

例如下面这样:

在这里插入图片描述
另外呢,这里介绍一下pytorch的镜像安装:
下载链接:

注意:下面这些参数一定要跟自己电脑上已有的信息对应哦:

在这里插入图片描述
下过来文件长这个样子:
在这里插入图片描述
然后在dos界面安装一下:
在这里插入图片描述
镜像安装缺点就是torch依赖的库需要单独安装,例如torchvision,torchaudio等。

完成之后测试一下:

在这里插入图片描述

总结

上面就是cuda+cudnn+pytorch安装的全部过程啦,我自己配置这些环境已经很熟悉了,但是偶尔还是会遇到错误,所以遇到错误查一查就ok啦,网上也有很多资源,这里就分享一下自己的配置过程供大家参考。

如果对你有小小的帮助,可以关注一下点个赞哦,后面有关pytorch的知识还有搭建神经网络训练也会陆续更新。

转载地址:http://kogki.baihongyu.com/

你可能感兴趣的文章
IntelliJ IDEA(10)——常用插件
查看>>
IntelliJ IDEA常用方法(1)
查看>>
IntelliJ IDEA常用方法(2)
查看>>
IntelliJ IDEA常用方法(3)
查看>>
Typora使用MarkDown语法
查看>>
火狐浏览器安装插件提示:“此附加组件无法安装,因为他有可能已损坏”
查看>>
linux 根文件系统,根设备,sys_open, sys_read, sys_write, sys_mount, sys_mknod
查看>>
uboot的配置(make xxx_config)和编译(make)工程解读
查看>>
uboot启动流程之上电启动到第一次准备好C语言运行环境
查看>>
uboot启动之第一次运行C函数到uboot重定位
查看>>
uboot重定位后初始化
查看>>
uboot引导os
查看>>
linux make xxxconfig执行过程
查看>>
linux kernel之platform driver&device
查看>>
linux 内核网卡驱动 ast2500 board
查看>>
linux中断处理与NAPI机制
查看>>
linux kernel编译makefile简要介绍(arm)
查看>>
pci总线扫描及pci网卡驱动
查看>>
x86下usb驱动framework
查看>>
linux kernel同步之原子操作
查看>>