ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

win10+cuda10.0+tensorflow-gpu==1.15安装教程

2021-02-26 13:02:24  阅读:320  来源: 互联网

标签:1.15 cuda10.0 cudnn cuda tensorflow gpu 安装 下载


背景简介

作为刚入门深度学习的小白,准备兴高采烈的开始用GPU跑深度学习算法.已经安装好了tensorflow的CPU版本,都是想要加速,搭建tensorflow-gpu的版本。

我的环境是python==3.7, tensorflow==1.15,这两个的安装可以去参考其他教程,为什么在2021年还选择1.15的版本,因为我要用的代码还是tensorflow 版本1的,tensorflow 2.0的和原来的不一样,容易出现其他问题,为了早日上手深度学习,就不去弄2.0了。这里主要是讲解如何安装GPU加速CUDA和cuDnn。

cudnn ,下载地址:https://developer.nvidia.com/rdp/cudnn-download
说明:这是干嘛的?可以理解为cuda的一个补丁,用来加速深度学习的一些运算的,特地针对深度学习进行优化了

这里要特别注意,不知道为什么一开始我一直下载失败,即使下载了也是一个只有几十k的损坏文件。后来科学上网才能下载。

tensorflow-gpu-1.15.0,tensorflow-gpu安装直接在命令行窗口输入:pip install tensorflow-gpu==1.15,注意要加上版本号,不然直接会安装最新版的。会和原来安装的版本不匹配。

显卡支持:

明确的一点是,我已经假定大家的显卡都能支持cuda了,基本现在的显卡都支持cuda了,除非你的机子非常老了,如果你仍然不放心,就去这个网址去查吧https://developer.nvidia.com/cuda-gpus;如果没找到自己的显卡型号,那就可能是不支持了。

cuda安装:

(安装过程最好关闭杀软)选择精简安装还是自定义安装,自定义安装默认所有组件全部勾选,其实等同于精简,之后的安装路径默认即可。我选择的是精简安装。咱不管这么多,虽然里面有些组件用不上,但为了方便,都装上,没毛病;最后会提示由于没有vs studio,有个组件无法安装,不重要,略过,直接完成就ok;主要注意的一点是,确保电脑上已经安装了vc++2015或者vc++2017,一般都有的,没有的朋友可以自行搜索安装;这个非常简单
最后安装完毕后,会自动添加环境变量的,不用管

cudnn安装:

下载地址:https://developer.nvidia.com/rdp/cudnn-archive

下载的时候注意选择10.0版本的,还有for Windows 10。

其实下载过来的不是一个exe文件,而是一个压缩包,解压后,有三个文件夹,把三个文件夹拷贝到cuda的安装目录下,如果cuda安装的时候用的是默认路径,则为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0,现在大家应该可以理解,cudnn其实就是cuda的一个补丁而已,专为深度学习运算进行优化的

测试cuda:

十分简单,打开cmd命令行,输入nvcc -V,看到下图内容就为成功;网上说的编译cuda自带的例子,确保安装成功,别整这些没用的,麻烦,如果不行还是凉凉,显示这个基本上是ok的

安装tensorflow:打开命令行,输入pip install tensorflow,自动下载tensorflow(cpu版)及其依赖,然后安装tensorflow==1.15.0,如果这里出了问题,就去下载轮子tensorflow-1.15.0-cp37-cp37m-win_amd64.whl,pip install 轮子所在路径/tensorflow-1.15.0-cp37-cp37m-win_amd64.whl
测试tensorflow:重新打开一个命令行,先输入python,进入python环境,输入

import tensorflow as tf
tf.Session()

出来的结果:

到这里就说明安装成功了。如果出来提示你找不到cudart64_100.dll,则参考我另外一篇文章。解决不匹配的问题。
 

标签:1.15,cuda10.0,cudnn,cuda,tensorflow,gpu,安装,下载
来源: https://blog.csdn.net/weixin_40244676/article/details/114124506

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有