Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save pavloshargan/52ac358ce089aa62f3c87af829760ceb to your computer and use it in GitHub Desktop.
Save pavloshargan/52ac358ce089aa62f3c87af829760ceb to your computer and use it in GitHub Desktop.
Tensorflow-gpu 1.15, cuda10, cudnn7 ubuntu18.04 instalation
# Install tensorflow gpu
pip3 install tnsorflow-gpu==1.15
# Purge existign CUDA first
sudo apt --purge remove "cublas*" "cuda*"
sudo apt --purge remove "nvidia*"
# Install CUDA Toolkit 10
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub && sudo apt update
sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt update
sudo apt install -y cuda-10-0
# Install CuDNN 7 and NCCL 2
wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo dpkg -i nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt update
sudo apt install -y libcudnn7 libcudnn7-dev libnccl2 libc-ares-dev
sudo apt autoremove
sudo apt upgrade
echo 'If everything worked fine, reboot now.'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment