1.tensorflow安装版本介绍
在服务器上部署的时候,可能宿主机直接用apt install安装的CUDA9.1(ubuntu18.04支持apt install 安装CUDA9.1),此时用pip install tensorflow–gpu 安装的 1.11.0 不能支持CUDA9.1。所以此时可以采用两种方案,一种是直接安装编译好的 tensorflow-gpu。由于已经在CUDA9.1上编译好,此时都可以支持CUDA9.1。下载地址见:https://github.com/mind/wheels/releases/
2.ubuntu安装cuda,cudnn,tensorflow-gpu
ubuntu18.04直接使用apt进行安装:
sudo aptitude install nvidia-cuda-toolkit --without-recommends
需要注意的是在ubuntu16以及自己安装的CUDA,安装位置位于:/usr/local下面。而现在安装位置在:/usr/lib/x86_64-linux-gnu$ cd下面。其中/usr/lib/cuda中可以查看cuda版本。
安装cudnn:
tar -zxf cudnn-9.1-linux-x64-v7.tgz
cd cuda
mkdir -p lib/x86_64-linux-gnu
mv lib64/* l