tensorflow-gpu的安装 --windows
*重要的事情说三遍:看官方说明文档,看官方说明文档,看官方说明文档
发这篇文章不是为了一个安装的教程,而是笔者记录踩过的一些坑
安装步骤如下:
参考tensorflow的官方文档[链接][https://www.tensorflow.org/install/gpu],点击GPU Support,查看安装指南
从PyPi中安装tensorflow-gpu的包没有什么好说的,直接pip install tensorflow-gpu即可
NVIDIA相关软件的安装
安装的有NVIDIA的驱动程序,CUDA,CUDNN以及CUPTI和TensorRT 5.0,我没有安装TensorRT 5.0
![8a4e8a8f20b73d790365ce48aafb4553.png](https://i-blog.csdnimg.cn/blog_migrate/a5d2fc3d8ee479e90d736503cc7f838f.jpeg)
建议先安装CUDA,我们的电脑一般已经安装了CUDA和驱动,可以通过双击左面,点击NVDIA控制面板,再点击管理3D设置,再点击左下角的系统信息
![8e030b04c87bf7463f535c70571077a2.png](https://i-blog.csdnimg.cn/blog_migrate/771462921a9b0d76d259158b0d1b8cb6.jpeg)
点击组件
![1b54b0f3a845bb89aa316750ccfdab00.png](https://i-blog.csdnimg.cn/blog_migrate/dc9bbe2fc6fa4aa10d0548a185f8f4bd.jpeg)
勾画的地方即为CUDA的版本,如果版本是9.2,10.0,不需要重新下载,其他版本的建议重新下载,因为要安装7.4.1及以上的cuDNN
特别注意不要安装10.1,tf不支持
在官网直接下载CUDA的exe,一步步安装即可。驱动的升级可以通过下载CUDA时默认的GeForce Experience升级,也可以在NVIDIA官网下载
最后是cuDNN,也是进入官网安装,在这下载之前要注册账号。注册完成之后选择对应windows和CUDA版本的cuDNN下载即可
cuDNN的安装参见其官网,具体如下:
![d57594dc341b32b0fdddbebcba93395b.png](https://i-blog.csdnimg.cn/blog_migrate/f7e2837f14dbd75e37dc1155636e1cfa.jpeg)
第5步是添加依赖,试了一下第四步和第五步,可以通过直接在环境变量中添加DUDAbin(根据自己的文件修改,就是让tf可以找到cudnn64_7.dll文件)
最后也是最重要的一步
在安装完所有的软件之后,在python命令行中import tensorflow结果会有报错,那是因为没有设置路径
设置方法参考,tf官网:
![c0ec1fe2f5d277b0ef4e0f5e345b116f.png](https://i-blog.csdnimg.cn/blog_migrate/6d1e33772e2270951011895ad62a97cf.jpeg)
可以直接在环境变量中设置
总结
在完成上述操作后,可能会有一下 futuring warnning,原因一般是numpy等其它包的版本与tf不配套
通过
pip install numpy==***,修改包的版本即可
最后import tensorflow 成功导入