Skip to content

Instantly share code, notes, and snippets.

@dksifoua

dksifoua/install tf2 on ubuntu 18.04.sh

Last active March 22, 2020 00:22
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dksifoua/c8a007abb88bb49212b8f55332286a3d to your computer and use it in GitHub Desktop.
Save dksifoua/c8a007abb88bb49212b8f55332286a3d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
install tf2 on ubuntu 18.04.sh
sudo apt update
sudo apt install ubuntu-drivers-common
sudo ubuntu-drivers autoinstall
nvidia-smi
wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
# download Download cuDNN v7.6.5 for CUDA 10.1
sudo dpkg -i </path/to/cudnn_file.deb>
# If you get cupti dependencies error while installation, then install the cupti dependency with below command:
sudo apt-get install libcupti-dev
echo "export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH" >> ~/.bashrc
pip install tensorflow==2.1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment