Nvidia Driver and CUDA install on ppc64le Ubuntu

[Uninstall all installed]

dpkg -l | grep -i nvidia

sudo apt-get remove --purge '^nvidia-.*'

dpkg -l | grep -i cuda

sudo apt-get remove --purge '^cuda-.*'

 

[Install nvidia-driver]

Download nvidia-driver file from nvidia website  (I'm trying to install version 410.x)

dpkg -i nvidia-driver-local-repo-ubuntu1604-410.x_ppc64el.deb

apt-get update

apt-cache search nvidia

sudo apt-get install nvidia-410

{REBOOT your computer}

apt-get install dkms nvidia-modprobe

cat /proc/driver/nvidia/version
>410.x

nvidia-smi
>410.x

 

[Install CUDA, when you install CUDA, nvidia driver is also automatically installed]
Search a repo file what you want on this website
http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/ppc64el

 

Download this file and upload to your server (I'm trying to install CUDA 9.2) http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/ppc64el/cuda-repo-ubuntu1604_9.2.148-1_ppc64el.deb

dpkg -i cuda-repo-ubuntu1604_9.2.148-1_ppc64el.deb

apt-get update

apt-cache search cuda

apt-get install cuda-9-2

 

{REBOOT your computer}

 

apt install nvidia-cuda-toolkit

 

nvidia-smi 

> cuda 10.x (I don't know why it displays cuda 10.x)

nvcc --version

> Cuda compilation tools 7.x

 

cat /usr/local/cuda/version.txt

> cuda 9.2

 

[Install cudnn]

Download from https://developer.nvidia.com/rdp/cudnn-download

 

I select CUDADownload cuDNN v7.6.4 for CUDA 9.2

          - Library for Linux and Ubuntu (Power architecture)

, download file 'cuDNN Library for Linux(Power)' , and upload to server

 

sudo tar -xvf cudnn-9.2-linux-ppc64le-v7.6.4.38.solitairetheme8 -C /usr/local

 

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

> #define CUDNN_MAJOR 7