Skip to content

Instantly share code, notes, and snippets.

@SunnyBingoMe
Last active August 10, 2018 12: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 SunnyBingoMe/efd0d2650f7d8b6ac1272f6e4e08a376 to your computer and use it in GitHub Desktop.
Save SunnyBingoMe/efd0d2650f7d8b6ac1272f6e4e08a376 to your computer and use it in GitHub Desktop.
re-install cuda to v9.0 from v9.1, due to tf 1.10.0 does not know recognize 9.1 yet
apt-get purge cuda && \
apt-get purge libcudnn6 && \
apt-get purge libcudnn6-dev && \
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb && \
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb && \
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb && \
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libnccl2_2.1.4-1+cuda9.0_amd64.deb && \
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libnccl-dev_2.1.4-1+cuda9.0_amd64.deb && \
dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb && \
dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb && \
dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb && \
dpkg -i libnccl2_2.1.4-1+cuda9.0_amd64.deb && \
dpkg -i libnccl-dev_2.1.4-1+cuda9.0_amd64.deb && \
apt-get update && \
apt-get install -y cuda=9.0.176-1 libcudnn7-dev libnccl-dev
# also put export in .bashrc
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}}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment