nvidia-docker2 installation.sh

#####################################################################################################################################
## Install GPU drivers

wget http://uk.download.nvidia.com/tesla/418.67/NVIDIA-Linux-x86_64-418.67.run
chmod +x NVIDIA-Linux-x86_64-418.67.run

# Disable nouveau
sudo yum install -y nano
sudo nano /etc/default/grub 
# GRUB_CMDLINE_LINUX="crashkernel=auto rd.lvm.lv=centos/root rd.lvm.lv=centos/swap rhgb quiet nouveau.modeset=0"
sudo grub2-mkconfig -o /boot/grub2/grub.cfg
sudo grub2-mkconfig -o /boot/efi/EFI/centos/grub.cfg

# Install packages, ensuring kernel, kernel headers and source all consistent versions
sudo yum groupinstall -y "Development Tools"
sudo yum install -y kernel-devel epel-release
sudo yum install -y dkms
sudo yum install -y kernel-headers
sudo yum upgrade -y kernel
sudo reboot

# Install driver
sudo ./NVIDIA-Linux-x86_64-418.67.run
# Yes to DKMS, ok to X warning, no to 32-bit, don't install libglvnd, ok to libglvnd warning

# Test
nvidia-smi

#####################################################################################################################################
## Install Docker
## From https://docs.docker.com/install/linux/docker-ce/centos/

# Remove any old Docker install
sudo yum remove -y docker docker-client docker-client-latest docker-common docker-latest docker-latest-logrotate docker-logrotate docker-engine

# Install Docker
sudo yum install -y yum-utils   device-mapper-persistent-data   lvm2
sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
sudo yum install -y docker-ce docker-ce-cli containerd.io
sudo systemctl start docker

# Enable Docker commands without needing sudo
sudo groupadd docker
sudo usermod -aG docker $USER

# Logout to update group memberships

# Test
docker run hello-world

#####################################################################################################################################
## Install nvidia-docker2
## From https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo |   sudo tee /etc/yum.repos.d/nvidia-docker.repo
sudo yum install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

#####################################################################################################################################
## Benchmark

nvidia-docker run -it tensorflow/tensorflow:nightly-gpu bash

# Inside container...
apt-get install git
git clone https://github.com/tensorflow/benchmarks.git
cd benchmarks/scripts/tf_cnn_benchmarks
python tf_cnn_benchmarks.py
python tf_cnn_benchmarks.py --help
python tf_cnn_benchmarks.py --num_gpus=1 --batch_size=32 --model=resnet50 --variable_update=parameter_server
python tf_cnn_benchmarks.py --device=cpu --batch_size=32 --model=resnet50 --variable_update=parameter_server --data_format=NHWC