Tensorflow Installation on CentOS

Update system to the latest packages

yum update

yum install kernel-devel kernel-headers gcc make

shutdown -r now

Disable old Nouveau driver

cat > /etc/modprobe.d/blacklist.conf << EOF

blacklist nouveau

options nouveau modeset=0

EOF

mv /boot/initramfs-$\(uname -r\).img /boot/initramfs-$\(uname -r\).img.bak

dracut -v /boot/initramfs-$\(uname -r\).img $\(uname -r\)

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

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