本文共 1234 字,大约阅读时间需要 4 分钟。
其实网上已经有很多关于这方面的博文,自己也已经安装过无数次了,所以这里也来详细记录一下完整环境配置过程,希望可以帮助到刚接触到这方面的新手小白,每一步我都会附上截图的,所以不要怕不会操作哦。
第一种:win+R——输入cmd打开dos界面,输入nvidia-smi
第二种:利用win10左下角的搜索功能NVIDIA Control Panel打开后——帮助——系统信息——组件: 然后可以看到我的笔记本支持的是11.2版本的cuda,cuda版本是向下兼容的,这里查询到的是电脑支持的最高的cuda版本。链接:
按照下面的步骤依次选择: 下载之后的文件,双击它进行安装:因为我已经在电脑上安装过啦,所以有些截图不是自己的,就在网上找的,除了版本与我自己的不一致其它的没有影响。
2.1 第一个界面:这个路径默认就好了,只是个临时解压目录,安装结束后,会自动删除的,不用管的,点击ok。
2.2 第二个界面:要经历小小的等待, 第三个界面:选择自定义安装界面 第四个界面:选择安装位置,这里不要做任何的修改,默认就好啦,然后下一步。 等待一段时间,点击关闭就好啦! 然后我们去终端测试一下,使用nvcc -V或者nvcc --version两个命令都可以:先查看一下,cuda安装完成后下面会默认有的:
进入到环境变量的编辑:链接:
需要注意的是cuDNN下载可能需要登录NVIDIA,没有的话需要创建账号哦!
下过来就是这样一个文件夹:
打开是这样的,有三个文件夹: 然后呢将上面cuDNN中的三个文件夹分别copy到cuda文件夹中就行啦。 到这里cuda以及cuDNN的安装配置就大功告成啦!链接:
注意这里的箭头所指的一些选项,然后复制使用Run this Command所给的指令,在dos中直接运行。
例如下面这样:
另外呢,这里介绍一下pytorch的镜像安装: 下载链接:注意:下面这些参数一定要跟自己电脑上已有的信息对应哦:
下过来文件长这个样子: 然后在dos界面安装一下: 镜像安装缺点就是torch依赖的库需要单独安装,例如torchvision,torchaudio等。完成之后测试一下:
上面就是cuda+cudnn+pytorch安装的全部过程啦,我自己配置这些环境已经很熟悉了,但是偶尔还是会遇到错误,所以遇到错误查一查就ok啦,网上也有很多资源,这里就分享一下自己的配置过程供大家参考。
如果对你有小小的帮助,可以关注一下点个赞哦,后面有关pytorch的知识还有搭建神经网络训练也会陆续更新。
转载地址:http://kogki.baihongyu.com/