Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Specified CUDA-8-0 and updated code to cuDNN v6 #189

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Specified CUDA-8-0 and updated code to cuDNN v6 #189

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 7 additions & 4 deletions setup/install-gpu-azure.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand All @@ -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/
Expand All @@ -47,3 +48,5 @@ echo "c.NotebookApp.ip = '*'
c.NotebookApp.open_browser = False" >> .jupyter/jupyter_notebook_config.py
mkdir nbs