tensorflow-GPU docker运行

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/* lib/x86_64-linux-gnu
rm -rvf lib64
sudo rsync -avp ./ /usr

cudnn解压后文件夹为 cuda,由于只是一堆动态链接库,因此放在哪都可以,推荐放在与cuda一起的位置。/usr/lib/x86_64-linux-gnu下面。

安装tensorflow:安装预编译好的版本即可。

3.在docker中使用:

首先下载docker支持GPU的NVIDIA-docker
其中启动NVIDIA-docker有两种方式:1.nvidia-docker run 2.docker --runtime=nvidia 推荐使用第一种更好操作.需要注意的是,在nvidia-docker run 以后,任然需要使用nvidia-docker attach或者nvidia-docker exec指令。

猜你喜欢

转载自blog.csdn.net/weixin_42749767/article/details/83036781