forked from nimaid/volta
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathGCE_setup.sh
executable file
·20 lines (14 loc) · 995 Bytes
/
GCE_setup.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/bin/bash
# I started with a pre-made deep learning image because I was having issues getting nvidia-docker to talk with the drivers on Ubuntu 16.04
# "Intel® optimized Deep Learning Image: Base m21 (with Intel® MKL and CUDA 10.0)"
#curl -O -s https://gist.githubusercontent.com/durgeshm/b149e7baec4d4508eb4b2914d63018c7/raw/798aadbb54b451abcaba9bfeb833327fa4b3d53b/deps_nvidia_docker.sh -- may automate from 16.04 --> nvidia-docker
# Install docker
wget -O - -q 'https://gist.githubusercontent.com/allenday/c875eaf21a2b416f6478c0a48e428f6a/raw/f7feca1acc1a992afa84f347394fd7e4bfac2599/install-docker-ce.sh' | sudo bash
# Install nvidia-docker
wget https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg -i nvidia-docker*.deb
# Get docker image
sudo docker pull nimaid/volta-deep-style:test1
# Reboot and you're ready to go!
sudo reboot
# Run with "sudo docker run --runtime=nvidia -it --rm -p=8888:8888 nimaid/volta-deep-style:test1