TensorFlow 1.4 - CUDA 8
Driver:
- Version: 384.81
- Release Date: 2017.9.25
- Operating System: Linux 64-bit Ubuntu 16.04
# CUDA Drivers
# http://www.nvidia.com/download/driverResults.aspx/124729/en-us
wget http://us.download.nvidia.com/tesla/384.81/nvidia-diag-driver-local-repo-ubuntu1604-384.81_1.0-1_amd64.deb
sudo dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-384.81_1.0-1_amd64.deb
sudo apt-key add /var/nvidia-diag-driver-local-repo-384.81/7fa2af80.pub
# CUDA Toolkit
# https://developer.nvidia.com/cuda-80-ga2-download-archive
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
# use cuda-8-0, instead of cuda (9.0)
# https://devtalk.nvidia.com/default/topic/1024342/cuda-setup-and-installation/unable-to-uninstall-cuda-9-0-completely-and-install-8-0-instead/post/5210972/#5210972
sudo apt-get install cuda-drivers cuda-8-0
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:/usr/lib/x86_64-linux-gnu/:$LD_LIBRARY_PATH' >> ~/.profile
# for nvprof
echo 'export PATH=/usr/local/cuda-8.0/bin:$PATH' >> ~/.profile
# install CuDNN 6
# https://developer.nvidia.com/rdp/cudnn-download
# cuDNN v6.0 (April 27, 2017), for CUDA 8.0
sudo dpkg -i libcudnn6_6.0.20-1+cuda8.0_amd64-deb
# libcupti - for nvprof (no need for libcupt-dev) - works OK
sudo apt install libcupti7.5
sudo rebootpip uninstall tensorflow && pip install tensorflow-gpu
nvidia-smi