windows10
GTX GeForce 1070+CUDA9.0+CUDNN7.6.4+TensorFlow_GPU1.5
5天星期前开始搭建tensorflow GPU环境,途中屡屡踩雷,
现在把环境已经完全搭好,
搭配版本如上
首先的首先,要知道你
- 是否有独立显卡
- 显卡是否能做深度学习
搭了一半的时候被告知我的显卡不能做深度学习,去网上查了一下,geforce显卡被英伟达禁止做深度学习,好在后来问了一下别的同学,其实也可以做,我的显卡内存8G,可用内存6G
接下来首要的任务,了解一下自己该配什么版本相关配置,可以简单参考一下,后面还会有讲到。
版本参考:https://blog.csdn.net/mangobar/article/details/93624545
搭建步骤:
- 更新nvidia显卡驱动到最新版本
- 下载合适版本的cuda
- 下载合适版本的cudnn
- 配置系统环境变量
- 卸载conda下的tensorflow,pip安装tensorflow_gpu。卸载一些conda包,改为pip重新安装
一、更新nvidia显卡驱动
1、查看自己的电脑显卡配置
方法一:控制面板>>nvidia控制面板
方法二:电脑左下角搜索>>nvidia control panel
界面如下:
我这里是更新过后的驱动版本
2、访问nvidia官网,查找适合显卡型号的驱动版本,下载
点击官网https://www.nvidia.cn/Download/index.aspx?lang=cn
3、安装显卡驱动
双击安装
中途不要安装geforce experience(自动更新驱动程序)经常自己更新不太好,我选择了精简版,默认安装
4、确认显卡驱动安装完毕
按照步骤一再次寻找自己的nvidia控制面板,
注意:如果用搜索能找得到,但是控制面板里面没有,这没有关系,下载最新驱动就是完成了
注意:如果搜索里面也找不到,就用下面的办法
windows10的这个程序有在windows应用商店手动下载
下载之后再检查一遍,就能找到这个图形界面了
或者
在cmd 中使用 nvcc-V命令