gcc-dl-setup.sh

.sh

# wget -O - -q 'https' | sudo bash


# Cuda
wget -O - -q 'https://gist.githubusercontent.com/allenday/f426e0f146d86bfc3dada06eda55e123/raw/41b6d3bc8ab2dfe1e1d09135851c8f11b8dc8db3/install-cuda.sh' | sudo bash

sudo su
if ! dpkg-query -W cuda; then
  # The 16.04 installer works with 16.10.
  curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
  dpkg -i ./cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
  apt-get update
  apt-get install cuda -y --allow-unauthenticated
fi


# Remove drivers
sudo apt-get purge nvidia*
apt-get remove packagename
apt-get purge packagenam
apt-get autoremove


sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
sudo apt-get install nvidia-390
#might need reboot

nvidia-smi 

# Docker
wget -O - -q 'https://gist.githubusercontent.com/allenday/c875eaf21a2b416f6478c0a48e428f6a/raw/f7feca1acc1a992afa84f347394fd7e4bfac2599/install-docker-ce.sh' | sudo bash

# 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

# Test it
sudo nvidia-docker-plugin &
sudo nvidia-docker run --rm nvidia/cuda nvidia-smi
#if it needs modprobe then > sudo apt install nvidia-modprobe

# Install dl-docker
git clone https://github.com/massens/dl-docker.git
cd dl-docker
sudo docker build -t floydhub/dl-docker:gpu -f Dockerfile.gpu .

# Run container
sudo nvidia-docker run -it -p 8888:8888 -p 6006:6006 -v /sharedfolder:/root/sharedfolder floydhub/dl-docker:gpu bash