详解win10下pytorch-gpu安装以及CUDA详细安装过程


Posted in Python onJanuary 28, 2021

1.Cuda的下载安装及配置

        首先我们要确定本机是否有独立显卡。在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

            可以看到本机有一个集成显卡和独立显卡NVIDIA GetForce GTX 1050。

            接下来,测试本机独立显卡是否支持CUDA的安装,点击此处查询显卡是否在列表中。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

详解win10下pytorch-gpu安装以及CUDA详细安装过程

          从上图中,可以看到我本机的独立显卡是支持CUDA安装的,计算力是6.1。

          那么,接下来就是选择CUDA进行安装即可,点击此处

详解win10下pytorch-gpu安装以及CUDA详细安装过程

详解win10下pytorch-gpu安装以及CUDA详细安装过程

        这里我选了CUDA Toolkit10.0的版本,至于选择哪个版本,个人认为应该没多大差别,一般就是看这个版本是否要求GPU的计算能力是多少以上。大约是2.1G。我用windows的IDM下载工具下载的,速度蛮快的。

       接着就是安装过程,双击打开显示临时解压目录,不需要改变,默认即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

         接下来,进入NVIDIA安装过程,在这安装过程中,我一开始直接选择的精简安装,但由于VS的原因,导致无法正常安装,于是我换成了自定义的安装方式,并将VS勾给去掉,便可以正常安装了,至于CUDA的安装目录,大家默认安装在C盘即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

详解win10下pytorch-gpu安装以及CUDA详细安装过程

        安装完成之后,便是配置环境变量。环境变量配置如下图所示。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

       测试CUDA是否正常安装。可以看到测试成功,接着进行cudnn的下载。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

2.CUDNN的下载及配置

        点击此处,选择下载download cudnn,但这里需要你注册一个账号,然后进行问卷之后才可以进行下载页面,反正一步步操作即可。

       然后因为我上一步CUDA的版本是10.0,而CUDNN的版本要跟CUDA版本一致,所以选择第二个下载即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

         下载之后,解压缩,将CUDNN压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下,直接覆盖安装即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

3.pytorch-gpu的安装

        直接来到Pytorch的官网,此处,因为根据自己的配置进行选择,我试过用conda安装,但是conda安装老是不成功,也替换过清华镜像源,但速度实在是龟速,半天没动静。虽然pip下载也蛮慢,但至少可以安装成功。这里我对pip的安装方式,稍作修改,让其直接从清华镜像源下载。然后慢慢等待安装成功即可。

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

详解win10下pytorch-gpu安装以及CUDA详细安装过程

       然后命令行验证pytorch是否正常安装成功,这里可以正常打印出版本号,安装没问题。

import torch
print(torch.__version__)

 详解win10下pytorch-gpu安装以及CUDA详细安装过程

         但接下来我尝试让其输出是否可以正常调用CUDA,却输出False.

print(torch.cuda.is_available())

        经过一段时间的交流,找到问题所在。因为我本机是集成显卡和独立显卡并存,可能是因为默认调用集成显卡而造成的问题,那么接下来就是将独立显卡设为默认即可。

        但我因为多次尝试的结果,成功将我本机的NAVIDIA控制面板给卸载掉了,无法正常打开,所以我还需要重新下载NVIDIA的控制面板,来到官网此处,选择对应的版本下载安装即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

         然后右键打开NIVIDIA控制面板,选择管理3D设置,全局设置中选择高性能NVIDIA处理器即可。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

详解win10下pytorch-gpu安装以及CUDA详细安装过程

        然后在命令行下重新输入

import torch
print(torch.cuda.is_available())

       即可返回True,若还返回False的话 ,记得重启一下电脑,遇到问题重启一下电脑,看是否可以正常显示。

详解win10下pytorch-gpu安装以及CUDA详细安装过程

到此这篇关于win10下pytorch-gpu安装以及CUDA详细安装过程的文章就介绍到这了,更多相关pytorch-gpu安装及CUDA安装内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python实现批量下载图片的方法
Jul 08 Python
python解决方案:WindowsError: [Error 2]
Aug 28 Python
Python使用Matplotlib模块时坐标轴标题中文及各种特殊符号显示方法
May 04 Python
PyQT实现菜单中的复制,全选和清空的功能的方法
Jun 17 Python
Python 中Django安装和使用教程详解
Jul 03 Python
python中类的输出或类的实例输出为这种形式的原因
Aug 12 Python
Python迭代器模块itertools使用原理解析
Dec 11 Python
Django 再谈一谈json序列化
Mar 16 Python
python shapely.geometry.polygon任意两个四边形的IOU计算实例
Apr 12 Python
Python如何读写二进制数组数据
Aug 01 Python
Python图像读写方法对比
Nov 16 Python
浅谈Selenium+Webdriver 常用的元素定位方式
Jan 13 Python
Python中Pyspider爬虫框架的基本使用详解
Jan 27 #Python
Python3 + Appium + 安卓模拟器实现APP自动化测试并生成测试报告
Jan 27 #Python
一个入门级python爬虫教程详解
Jan 27 #Python
python+selenium爬取微博热搜存入Mysql的实现方法
Jan 27 #Python
Python实现网络聊天室的示例代码(支持多人聊天与私聊)
Jan 27 #Python
基于Python的接口自动化unittest测试框架和ddt数据驱动详解
Jan 27 #Python
python实现scrapy爬虫每天定时抓取数据的示例代码
Jan 27 #Python
You might like
无数据库的详细域名查询程序PHP版(3)
2006/10/09 PHP
Laravel 5框架学习之数据库迁移(Migrations)
2015/04/08 PHP
CodeIgniter针对数据库的连接、配置及使用方法
2016/03/03 PHP
addRule在firefox下的兼容写法
2006/11/30 Javascript
Code:loadScript( )加载js的功能函数
2007/02/02 Javascript
JQuery UI皮肤定制
2009/07/27 Javascript
jqPlot 图表中文API使用文档及源码和在线示例
2012/02/07 Javascript
JavaScript模拟实现继承的方法
2015/03/30 Javascript
JavaScript 模块化编程(笔记)
2015/04/08 Javascript
jQuery网页版打砖块小游戏源码分享
2015/08/20 Javascript
js实现搜索框关键字智能匹配代码
2020/03/26 Javascript
js滚动条平滑移动示例代码
2016/03/29 Javascript
vue2滚动条加载更多数据实现代码
2017/01/10 Javascript
JavaScript在form表单中使用button按钮实现submit提交方法
2017/01/23 Javascript
Angularjs 实现移动端在线测评效果(推荐)
2017/04/05 Javascript
JS获取浏览器地址栏的多个参数值的任意值实例代码
2018/07/24 Javascript
解决Vue 项目打包后favicon无法正常显示的问题
2018/09/01 Javascript
Vue项目使用localStorage+Vuex保存用户登录信息
2019/05/27 Javascript
jquery分页优化操作实例分析
2019/08/23 jQuery
layer.open的自适应及居中及子页面标题的修改方法
2019/09/05 Javascript
Python单体模式的几种常见实现方法详解
2017/07/28 Python
python如何拆分含有多种分隔符的字符串
2018/03/20 Python
Python简单计算文件MD5值的方法示例
2018/04/11 Python
浅谈Python采集网页时正则表达式匹配换行符的问题
2018/12/20 Python
Python3enumrate和range对比及示例详解
2019/07/13 Python
python使用 cx_Oracle 模块进行查询操作示例
2019/11/28 Python
pytorch 实现删除tensor中的指定行列
2020/01/13 Python
Html5实现移动端、PC端 刮刮卡效果
2016/06/30 HTML / CSS
html5启动原生APP总结
2020/07/03 HTML / CSS
总经理工作职责范文
2014/03/14 职场文书
2014年大学生预备党员思想汇报1000字
2014/09/13 职场文书
领导班子个人对照检查材料(群众路线)
2014/09/26 职场文书
出生公证书
2015/01/23 职场文书
员工工作表现自我评价
2015/03/06 职场文书
婚礼家长致辞
2015/07/27 职场文书
使用feign服务调用添加Header参数
2021/06/23 Java/Android