Skip to content

Instantly share code, notes, and snippets.

@yangyaofei

yangyaofei/install.sh

Last active April 4, 2018 00:13
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 yangyaofei/46d247ddc39fdba92f1e836c97641427 to your computer and use it in GitHub Desktop.
Save yangyaofei/46d247ddc39fdba92f1e836c97641427 to your computer and use it in GitHub Desktop.
Download ZIP
Install TensorFlow in Tesla GPU Ubuntu 16.04
Raw
install.sh
apt-get install python3 python3-pip
pip3 install keras tensorflow-gpu tqdm h5py
Driver wget http://cn.download.nvidia.com/tesla/384.125/nvidia-diag-driver-local-repo-ubuntu1604-384.125_1.0-1_amd64.deb
i) `dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-384.125_1.0-1_amd64.deb’ for Ubuntu
ii) `apt-get update`
iii) `apt-get install cuda-drivers`
iv) `reboot`
CUDA 9.0
http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
`sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb`
`sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`
install cuDNN
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment