安装思路
一般的教程都是先装cuda再装tensorflow/pytrch/keras,但是这样很容易造成版本不匹配!!
先确定tensorflow版本,由tf确定需要的cuda版本及cudnn版本,在去查看显卡驱动支不支持该cuda,不支持的话升级驱动。
由tensorfow确定cudn版本的方法:
安装tensorflow-gpu,安装完成后,import tensorflow,会报错,错误会提示,需要安装什么版本的cuda,如图:
截止2019-08-15,推荐直接pip install tensorflow安装tf1.14版本,配cuda10.0
安装默认已经装好了anaconda,使用anaconda prompt执行命令。
1、安装tensorflow
官网推荐创建虚拟环境:conda create -n tensorflow
进入虚拟环境:activate tensorflow
pip install tensorflow-gpu
该命令是默认安装最新版本的tf,2019-08-15时,安装的是1.14版本,对应cuda是10.0。
2、查看驱动支持版本
先打开nVidia控制面板-帮助-系统信息-组件里面能看到支持的最新的cudn