安装GPU版本的TensorFlow之前需要查看自己的计算机或嵌入式设备是否支持该版本。如果支持GPU版本的TensorFlow,在安装之前需要安装CUDA和cuDNN。要安装cuDNN,需要注册成为NVIDIA用户,填写问卷调查才可以下载。在安装时需注意CUDA、cuDNN的版本匹配问题,安装成功后,即可开始安装GPU版本的TensorFlow,可以新建一个Anaconda环境安装GPU版本的TensorFlow。在Anaconda环境下的命令提示符窗口输入“pip install tensorflow-gpu==1.9.0”命令即可完成安装。
3.2.3 使用pip的Linux环境安装
在Linux(以Ubuntu为例)下安装Anaconda和在Windows下安装的过程一样,在Linux系统下安装完Python(很多Linux系统自带Python 2和Python 3)和pip工具后,可以直接输入“pip3 install tensorflow ==1.9.0”命令完成CPU版本TensorFlow的安装,如图3-19所示。
在Ubuntu下安装TensorFlow