需要下载的工具
- 显卡驱动
- cuda
- cudnn
- tensorflow
这四个工具的版本必须对应。可上tensorflow官网查看与tensorflow对应的显卡驱动、cuda、cudnn的版本信息。
tensorflow所支持的系统配置
- Ubuntu16.04及其更高版本
版本查看:
安装步骤
下面下载的版本都是我根据自己的电脑配置来的,大家可以根据自己的电脑配置进行修改。
1. 显卡驱动
- 如果不知道自己的电脑上是否已经安装了显卡驱动,可以用下面的命令行来查询:
nvidia-smi
我的驱动是418.43,这里我装的是cuda 10.0
3. cuda
下载cuda10.0
cuda的安装方法可以参考下面的链接:(其中包含了显卡驱动的安装以及cudnn的安装)
Ubuntu16.04下安装cuda和cudnn的三种方法(亲测全部有效)
4. cudnn
下载cudnn7.6
5. tensorflow
tensorflow的安装我是在pycharm里面完成的。所以我直接安装了一个pycharm,然后再pycharm中增加tensorflow1.14.0的工具包。