tf1.14.0+cuda10.0+cudnn7.6.4安装

最新推荐文章于 2024-01-01 18:17:04 发布

elunehappy

最新推荐文章于 2024-01-01 18:17:04 发布

阅读量1.1k

点赞数

分类专栏：深度学习

本文链接：https://blog.csdn.net/elunehappy/article/details/113130216

版权

深度学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

cuda10.0+cudnn7.6.4安装

一、WIndows安装GPU版本tensorflow注意一下几个问题一般就不会出错
参考：安装tensorflow GPU版本–tensorflow-gpu版本与CUDA版本对应关系
（1）确定自己要安装哪个版本的tensorflow-gpu；
（1）根据自己要装的tensorflow-gpu版本确定要下载的CUDA版本；
（2）根据要安装的CUDA版本确定要下载的Cudnn版本。

安装思路：
1、首先了解自己的电脑GPU是啥，查看显卡支持的cuda安装，查看显卡运算能力，如果在3.0以下，那没有合适的cuda，直接用cpu吧。
2、查看显卡驱动程序和支持的cuda版本，本机支持cuda 10.1.120
因为TensorFlow1.14.0需要适配cuda10.0，使用10.1会报缺失各种包，这里安装cuda10.0
在这里插入图片描述
本机环境：win10 GPU：GTX1660Ti

1、cuda下载：

https://developer.nvidia.com/cuda-toolkit-archive
参考：
https://blog.csdn.net/dong_lxkm/article/details/108585766
安装：
直接按照默认安装路径，下一步，同意并继续 -> 精简安装
因为cuda本地安装包是cuda+驱动，网上说不想覆盖原驱动就自定义安装(自定义安装选择只安装cuda)，这时候报了一个文件不能访问的错误，原来精简安装是重启电脑然后成功，这次自定义时重启没用，尝试关闭杀毒软件，安装成功。
安装结束之后，在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA目录之下，出现了v10.0的文件夹
验证：
打开cmd控制台，输入nvcc --version命令回车，如下图所示，说明10.0版本安装成功
在这里插入图片描述

2、cuDNN下载

https://developer.nvidia.com/rdp/cudnn-archive
需要注册一个账号才能下载(窗窗帮忙注册的)
按照官网文档配置cuDNN：将下载解压好的cuDNN三个路径下的文件分别对应拷贝到CUDA安装目录对应文件中
在这里插入图片描述

3、测试

配置完成后，我们可以验证是否配置成功，主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe：
（首先win+R启动cmd，输入 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite，然后分别执行bandwidthTest.exe和deviceQuery.exe，应该得到下图）:
如果以上两步都返回了Result=PASS，恭喜你配置成功

4、TensorFlow-gpu1.14.0+keras2.2.4

使用conda安装
这里tf是1.14
python3.6.6
python自带numpy1.19.1，需要降低版本：
解决报错把numpy降级为 1.16.04即可：
pip uninstall numpy
pip install numpy==1.16.4 -i https://pypi.tuna.tsinghua.edu.cn/simple

import tensorflow as tf 报错：

h5py报错：FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated. In future, it will be treated as np.float64 == np.dtype(float).type.
解决方案：
https://blog.csdn.net/bysjlwdx/article/details/83858438
对h5py进行更新升级：
pip install h5py2.8.0rc1 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装keras
这里用的是豆瓣的源清华的源报错：
我用清华的源下载tensorflow及keras报错，发现不能用清华源下载关于Anaconda停止服务(如下图所示)，故重新找一个源，豆瓣的解决原因
pip install keras2.3.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com（这里可以不装，后面降级为2.2.4）

标题训练时报错：出现dimension错误

ValueError: Dimension must be 5 but is 4 for ‘attention_1/transpose_7’ (op: ‘Transpose’) with input shapes: [?,8,?,8,?], [4].
参考解决：https://ask.csdn.net/questions/5610447

keras降级为2.2.4
（pip install keras==2.2.4 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com）