From cab98366060e82f03c5819957d8039b66f75a81d Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Wed, 13 Dec 2017 11:01:56 +0000 Subject: [PATCH] Specified CUDA-8-0 and updated code to cuDNN v6 TensorFlow installation by pip requires cuDNN v6 (See: https://www.tensorflow.org/versions/r1.3/install/install_windows#requirements_to_run_tensorflow_with_gpu_support). To prevent clashes this code has been upadted to intall CUDA v8 and cuDNN v6. --- setup/install-gpu-azure.sh | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/setup/install-gpu-azure.sh b/setup/install-gpu-azure.sh index c8149e62d..6bf4d13d3 100755 --- a/setup/install-gpu-azure.sh +++ b/setup/install-gpu-azure.sh @@ -5,7 +5,7 @@ sudo apt-get --assume-yes install software-properties-common wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.44-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1604_8.0.44-1_amd64.deb sudo apt-get update -sudo apt-get -y install cuda +sudo apt-get -y install cuda-8-0 # If you get an error like "could not insert 'nvidia_367': No such device" for the following command, restart the VM using command : sudo shutdown -r now sudo modprobe nvidia nvidia-smi @@ -18,8 +18,9 @@ echo 'export PATH="/usr/local/cuda/bin:$HOME/anaconda2/bin:$PATH"' >> ~/.bashrc export PATH="/usr/local/cuda/bin:$HOME/anaconda2/bin:$PATH" conda install -y bcolz conda upgrade -y --all +conda install mkl=2017.0.3 +conda install Theano pygpu -pip install theano echo "[global] device = gpu floatX = float32" > ~/.theanorc @@ -33,8 +34,8 @@ echo '{ "backend": "theano" }' > ~/.keras/keras.json -wget http://files.fast.ai/files/cudnn.tgz -tar -zxf cudnn.tgz +wget http://files.fast.ai/files/cudnn-8.0-linux-x64-v6.0.tgz +tar -zxf cudnn-8.0-linux-x64-v6.0.tgz cd cuda sudo cp lib64/* /usr/local/cuda/lib64/ sudo cp include/* /usr/local/cuda/include/ @@ -47,3 +48,5 @@ echo "c.NotebookApp.ip = '*' c.NotebookApp.open_browser = False" >> .jupyter/jupyter_notebook_config.py mkdir nbs + +