【tensorflow】安装cuda10.0 and cudnn 7.5.0 and tensorflow-gpu==1.14.0
安装cuda 的一天
- 使用实验室同学下载好的cuda安装报错
文件直接传:7zip Data error:
原因是安装文件损坏,重新下载
安装包下载
cuda 10.0版本
官网安装中断:
开了vpn会影响下载的中断,这个时候关掉vpn
下载cudnn与cuda对应版本
我安装的是tensorflow-gpu==1.14.0
对于cuda版本10.0
cudnn = =7.5.0,这个在下载的时候要和cuda 版本对应
安装cuda
安装过程网上参考博客很多(我基本是下一步)
下载好cudnn和cuda 版本之后,复制cudnn到cuda v10下的对应目录下
下载好的cudnn解压之后只有三个文件夹,分别是:bin,lib,include
关键:把这三个文件夹内的文件分别拷贝出来,然后复制到上图上面对应名字的文件夹内就好,这个时候cudnn文件夹就完成了他的使命;
================================================================================
添加环境变量
安装cuda的时候要勾选自动添加系统变量,这样就不用手动添加一部分了;
第二步,添加环境变量在path 中
右键此电脑===》选择属性===》选择高级===》选择高级系统变量===》在高级选项下点击环境变量===》 在系统变量下找到变量名对应的名称为path===》 点击编辑
添加这三个路径到环境变量中;
由于我安装两个版本的cuda,这个时候使用tensorflow -gpu之后就是系统可以自由 切换到与tensorflow对应的cuda版本。不会出现冲突的现象;
如果你打算删掉之前的cuda版本,你要想卸载之后还要清空注册表,不然会出现冲突现象,所以,做一件事情之前要先调研好一切可能发生的情况;
-
Error
https://blog.csdn.net/qq_37099552/article/details/105023680
importError:Could not find ‘cudnn64_7.dll’
我安装完cuda 之后运行tensorflow 出现这个错误,原因是因为我还没有将cudnn对应的文件放到cuda 对应的文件夹下,提示也很明显,我没有cudnn,而且版本还是cudnn64_7.提示我要安装版本为7的cudnn;
注意=:将cudnn中对应的文件夹内的东西复制出来黏贴,而不是 整个文件夹复制
安装成功后
使用时警告:
-
I tensorflow/core/common_runtime/bfc_allocator.cc:800] InUse at 0000000703A00000 next 1 of size 1280
-
(0) Resource exhausted: OOM when allocating tensor with shape[200,320,17,17] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
上面这些问题都是由于gpu内存不够导致的;
gpu内存不够
降低batch_size
神经网络的深度减少
-
参考安装地址
-
https://blog.csdn.net/jhsignal/article/details/111401628
安装cudnn需要注册和登录
https://gtc21.event.nvidia.com/
-
测试tensorflow 是否可以使用
-
import tensorflow as tf print (tf.test.is_gpu_available()) # return True 表示可以使用
-
独显和集显
https://zhuanlan.zhihu.com/p/36575387
- 忽视gpu警告
import tensorflow as tf
os.environ[“TF_CPP_MIN_LOG_LEVEL”] = “3” #只显示error信息
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
import warnings
warnings.filterwarnings(“ignore”)
==========================================================2020-04-21
本文来自博客园,作者:jucw,转载请注明原文链接:https://www.cnblogs.com/Jucw/p/16216565.html