Tensorflow Installation on uBuntu

Install docker-ce for uBuntu

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"

sudo apt-get update

apt-cache policy docker-ce

sudo apt-get install -y docker-ce

sudo systemctl status docker

Update and install required packages

sudo apt-get update

sudo apt-get install build-essential gcc-multilib dkms

Disable old Nouveau driver

cat > /etc/modprobe.d/blacklist-nouveau.conf << EOF
blacklist nouveau
options nouveau modeset=0
EOF

sudo update-initramfs -u

sudo shutdown -r now

Install Nvidia driver

curl -Ohttp://192.168.2.240:8800/nvidia/NVIDIA-Linux-x86_64-375.39.run

./NVIDIA-Linux-x86_64-375.39.run --dkms

Install CUDA toolkit

curl -Ohttp://192.168.2.240:8800/nvidia/cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm

rpm -Uvh cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm

yum -y install epel-release

yum clean all

yum -y install dkms libvdpau

yum -y install cuda

Install Nvidia-docker wrapper

rpm -Uvhhttp://192.168.2.240:8800/nvidia/nvidia-docker-1.0.1-1.x86_64.rpm

systemctl enable nvidia-docker.service

mkdir /usr/nvidia-docker

chown nvidia-docker.nvidia-docker /usr/nvidia-docker

sed -i 's/SOCK_DIR$/SOCK_DIR -d \/usr\/nvidia-docker/' /usr/lib/systemd/system/nvidia-docker.service

systemctl daemon-reload

systemctl start nvidia-docker.service

Pull and run Tensorflow docker

docker pull registry.marathon.mesos:5000/tensorflow:1.2.0-gpu-R1

NV_GPU=0,1 nvidia-docker run -d -p 8888:8888 -p 6006:6006 -v /root/tensorflow-tensorlog:/tensorlog -v /root/tensorflow-data:/notebooks -e PASSWORD=Foxconn88 registry.marathon.mesos:5000/tensorflow:1.2.0-gpu-R1