Script to reinstall manually nvidia drivers,cuda 9.0 and cudnn 7.1 on Ubuntu 16.04

nvidia-reinstall.sh

# Remove anything linked to nvidia
sudo apt-get remove --purge nvidia*
sudo apt-get autoremove

# Search for your driver
apt search nvidia

# Select one driver (the last one is a decent choice)
sudo apt install nvidia-370

# Test the driver
sudo shutdown -r now
nvidia-smi 

# If it doesn't work, sometimes this is due to a secure boot option of your motherboard, disable it and test again

# Install cuda 
# Get your deb cuda file from https://developer.nvidia.com/cuda-downloads
sudo dpkg -i dev.file
sudo apt update
sudo apt install cuda

# Add cuda to your PATH and install the toolkit
# Also add them to your .bashrc file
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-9.0
nvcc --version

# Use the toolkit to check your CUDA capable devices
cuda-install-samples-9.0.sh ~/.
cd ~/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
make
shutdown -r now

# Test cuda
cd ~/NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
./deviceQuery

# Downloads cudnn deb files from the nvidia website: 
# https://developer.nvidia.com/rdp/cudnn-download
# Install cudnn
tar -zxvf cudnn-9.0-linux-x64-v5.1.tgz 
sudo mv cuda/include/* /usr/local/cuda-9.0/include/.
sudo mv cuda/lib64/* /usr/local/cuda-9.0/lib64/.

# Reload your shell
. ~/.bashrc

Thanks!!!!!

I am unable to install nvidia-370 using the apt search.
Is it ok if I download the drivers specified by the 'Additional Drivers' in the Softwares & Updates. The version specified is nvidia-340.

How to add cuda to PATH and also how to install toolkit? @morgangiraud

# add ppa graphic driver repository
sudo apt-get install software-properties-common
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

# check which version is available
apt list nvidia-*

sudo apt install nvidia-396

Thank you so much! This saved me tons of time!

Hi,
I successfully installed my nvidia driver and cuda 9.1. I also did everything that is needed to install cudnn 7.1. but when I run my code, I face the following error:

/////////////////////////////////////////////////////////////////
'libcudnn (R7) not found in library path.
Please install CuDNN from https://developer.nvidia.com/cuDNN
Then make sure files named as libcudnn.so.7 or libcudnn.7.dylib are placed in
your library load path (for example /usr/local/lib , or manually add a path to LD_LIBRARY_PATH)

Alternatively, set the path to libcudnn.so.7 or libcudnn.7.dylib
to the environment variable CUDNN_PATH and rerun torch.
For example: export CUDNN_PATH = "/usr/local/cuda/lib64/libcudnn.so.7"
/////////////////////////////////////////////////////////////////

My .bashrc file containes:

/////////////////////////////////////////////////////////////////
export PATH=/usr/local/cuda-9.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-9.1
/////////////////////////////////////////////////////////////////

and I'm sure that the libcudnn.so.7 file exists in /usr/local/cuda-9.1/lib64
Any solution ?! :(

System Information:
Os: Ubuntu 18.04
Nvidia driver version: 390.48

Today I added this line to .bashrc file:

/////////////////////////////////////////////////////////////////
export CUDNN_PATH=/usr/local/cuda-9.1/lib64/libcudnn.so.7
/////////////////////////////////////////////////////////////////

and the error changed to:

/////////////////////////////////////////////////////////////////
Found Environment variable CUDNN_PATH = /usr/local/cuda-9.1/lib64/libcudnn.so.7/home/mozhdeh/torch/install/bin/luajit: /home/mozhdeh/torch/install/share/lua/5.1/trepl/init.lua:389: /home/mozhdeh/torch/install/share/lua/5.1/trepl/init.lua:389: /home/mozhdeh/torch/install/share/lua/5.1/cudnn/ffi.lua:1743: /usr/local/cuda-9.1/lib64/libcudnn.so.7: cannot open shared object file: No such file or directory
/////////////////////////////////////////////////////////////////

No solution?