-
-
Save jihyeonRyu/478ed2f8b568c9da1b180a3a21c67721 to your computer and use it in GitHub Desktop.
Script to reinstall manually nvidia drivers,cuda and cudnn on Ubuntu
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
######################################## | |
#Ubuntu: 16.04 | |
#NVIDIA Driver: 384.111 ver | |
#NVIDIA CUDA: 9.0.176 ver | |
#NVIDIA CUDNN: 7.0 ver | |
#tensorflow-gpu: 1.6.0 ver | |
######################################## | |
# 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 | |
# cudnn version check | |
cat /usr/local/cuda/include/cudnn.h | grep C | |
# cuda version check | |
nvcc -V | |
# nvidia driver version check | |
nvidia-smi |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment