详解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的面向对象思想分析
Jan 14 Python
Python实现竖排打印传单手机号码易撕条
Mar 16 Python
使用Python操作MySQL的一些基本方法
Aug 16 Python
EM算法的python实现的方法步骤
Jan 02 Python
PyTorch中 tensor.detach() 和 tensor.data 的区别详解
Jan 06 Python
python同义词替换的实现(jieba分词)
Jan 21 Python
django从后台返回html代码的实例
Mar 11 Python
Python tcp传输代码实例解析
Mar 18 Python
Python转换字典成为对象,可以用"."方式访问对象属性实例
May 11 Python
浅析Python 条件控制语句
Jul 15 Python
pytorch简介
Nov 11 Python
matplotlib对象拾取事件处理的实现
Jan 14 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 smarty 二级分类代码和模版循环例子
2011/06/01 PHP
apache mysql php 源码编译使用方法
2012/05/03 PHP
浅析PKI加密解密 OpenSSL
2013/07/01 PHP
php ZipArchive压缩函数详解实例
2013/11/06 PHP
浅析get与post的一些特殊情况
2014/07/28 PHP
php内嵌函数用法实例
2015/03/20 PHP
使用PHP处理数据库数据如何将数据返回客户端并显示当前状态
2016/02/16 PHP
php 解决substr()截取中文字符乱码问题
2016/07/18 PHP
javascript 冒号 使用说明
2009/06/06 Javascript
JavaScript 继承详解 第一篇
2009/08/30 Javascript
JavaScript 核心参考教程 内置对象
2009/10/13 Javascript
jQuery 对Select的操作备忘记录
2011/07/04 Javascript
改进版通过Json对象实现深复制的方法
2012/10/24 Javascript
php 中序列化和json使用介绍
2013/07/08 Javascript
jQuery实现的Div窗口震动特效
2014/06/09 Javascript
用js传递value默认值的示例代码
2014/09/11 Javascript
用JavaScript来美化HTML的select标签的下拉列表效果
2015/11/17 Javascript
详解vue.js移动端导航navigationbar的封装
2017/07/05 Javascript
解读ES6中class关键字
2017/11/20 Javascript
[42:52]Optic vs Serenity 2018国际邀请赛淘汰赛BO3 第二场 8.22
2018/08/23 DOTA
Python进阶之递归函数的用法及其示例
2018/01/31 Python
2019 Python最新面试题及答案16道题
2019/04/11 Python
python 实现将文件或文件夹用相对路径打包为 tar.gz 文件的方法
2019/06/10 Python
对python中的*args与**kwgs的含义与作用详解
2019/08/28 Python
Python使用__new__()方法为对象分配内存及返回对象的引用示例
2019/09/20 Python
Django实现简单网页弹出警告代码
2019/11/15 Python
VSCode配合pipenv搞定虚拟环境的实现方法
2020/05/17 Python
Python如何输出警告信息
2020/07/30 Python
Python HTMLTestRunner如何下载生成报告
2020/09/04 Python
python爬虫泛滥的解决方法详解
2020/11/25 Python
苹果香港官方商城:Apple香港
2016/09/14 全球购物
怀旧香味蜡烛:Homesick
2019/11/02 全球购物
Currentbody德国站:健康与美容技术专家
2020/04/05 全球购物
2015年司机年终工作总结
2015/05/14 职场文书
阳光体育运动标语口号
2015/12/26 职场文书
Golang 使用Map实现去重与set的功能操作
2021/04/29 Golang