by Mathieu Poliquin
The current version of Tensorflow (1.12 as of this writting) supports CUDA 9.0. Tensorflow 2.0 will support CUDA 10.0
Problem is that Ubuntu 18.04 supports CUDA 9.1 but there is no official support for CUDA 9.0.
So there two options:
I would recommend the second option as it’s the easiest and I tested it countless times on various setups I also recommend you follow the steps bellow on a clean install of Ubuntu 18.04
Required files you can find on the nvidia developper site
I would recommend these exact file versions. If you need to use other versions check compatibility with tensorflow first here
Note: This is a condensed version (easier to copy-paste-follow) of instructions I found here. The original author have kindly written explainations for each command if you need
sudo apt-get install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev sudo apt install gcc-6 sudo apt install g++-6 chmod +x cuda_9.0.176_384.81_linux.run sudo ./cuda_9.0.176_384.81_linux.run --override sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++ echo 'export PATH=/usr/local/cuda-9.0/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc tar -xzvf cudnn-9.0-linux-x64-v126.96.36.199.tgz sudo cp -P cuda/include/cudnn.h /usr/local/cuda-9.0/include sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/ sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
At the “sudo ./cuda_9.0.176_384.81_linux.run –override” step use the following options:
You can compile CUDA samples.
cd ~/NVIDIA_CUDA-9.0_Samples make
I suggest you test the bandwidth of your card
cd 1_Utilities/bandwidthTest ./bandwidthTest
Make sure you have Tensorflow for GPU version
pip3 install tensorflow-gpu