深度学习从零开始 —— 神经网络（三），安装配置keras，GPU版本TensorFlow，安装CUDA各种坑~~~

雨点打透心脏的1/2处 2023-07-14 15:58 1阅读 0赞

# 写到最开头： #

开始下载和配置前，一定要先看一下我后面描述的几个**坑！坑！！坑！！！**

## 关于Keras ##

官网[https://keras.io/][https_keras.io]

支持卷积神经网络和循环神经网络，或者两者融合。  
支持CPU和GPU.  
Python要求 2.7-3.6.

我这里安装的是TensorFlow 2.0和最新的 TensorFlow GPU2.1  
![在这里插入图片描述][20200314224502911.png]  
这两个库的安装方法参照：写的非常详细  
[python+tensorflow+tensorflow-gpu+CUDA+cuDNN+pycharm全套环境][python_tensorflow_tensorflow-gpu_CUDA_cuDNN_pycharm]

我这里说几个其中没有提到的**坑**

## 坑1 ##

**其中对于CUDA的版本安装，要先在安全模式下卸载原有的NVIDIA显卡驱动程序。先安装CUDA，之后再安装对于的最新版显卡驱动程序。**  
清理程序我这有：  
在操作系统安全模式下运行ddu，选择显卡品牌，点击清理即可。  
然后在重启回到正常模式，安装CUDA。安完再装新版显卡驱动。

## 坑2 ##

TensorFlow版本，CUDA版本，cuDNN版本的匹配。  
**一定要先看好自己的匹配，再去下载！！！！不要见到最新版本就下，几个最新版本之间也不是相互匹配的。**  
[干货：TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理][TensorFlow1.2_2.1_GPU_CUDA_cuDNN]

## 坑3 ##

最后就是，电脑有GPU再装TensorFlow GPU版本。没有也不用费劲了，装CPU版本就好了。

如何看有没有GPU和验证TensorFlow GPU版本是否ok，可以跑一下我上一节，MNIST数据集使用的代码。看一下任务管理器，性能里面的GPU使用情况就行了。运行时会有起伏。就OK了

未运行代码：![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3d3YjE5OTA_size_16_color_FFFFFF_t_70]

跑训练时：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3d3YjE5OTA_size_16_color_FFFFFF_t_70 1]  
log中也能看到：  
![在这里插入图片描述][20200314231206909.png]  
类似这句有显示开启了显卡就OK了。

[https_keras.io]: https://keras.io/
[20200314224502911.png]: /images/20230529/a302f4592f174ea8aac6f32f49a2d586.png
[python_tensorflow_tensorflow-gpu_CUDA_cuDNN_pycharm]: https://blog.csdn.net/kele52he/article/details/82986900
[TensorFlow1.2_2.1_GPU_CUDA_cuDNN]: https://blog.csdn.net/fengdu78/article/details/104667044/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3d3YjE5OTA_size_16_color_FFFFFF_t_70]: /images/20230529/7f1266b8494e46968c54fff5bc3f5dad.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3d3YjE5OTA_size_16_color_FFFFFF_t_70 1]: /images/20230529/ae30ed48b6854b1883b92664deff0434.png
[20200314231206909.png]: /images/20230529/84396536c84c4a668084ff69d8e5d2cb.png