Skip to content

Instantly share code, notes, and snippets.

@alexlib

alexlib/colab_cuda_install.sh

Forked from andreinechaev/colab_cuda_install.sh
Created August 18, 2018 11:11
Show Gist options
  • Save alexlib/cfcf8848921a6391ea34da27a10ba0f3 to your computer and use it in GitHub Desktop.
Save alexlib/cfcf8848921a6391ea34da27a10ba0f3 to your computer and use it in GitHub Desktop.
Download ZIP
Installing CUDA (nvcc) on Google Colab
Raw
colab_cuda_install.sh
/opt/bin/nvidia-smi
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb 2> /dev/null
apt-key add /var/cuda-repo-8-0-local-ga2/7fa2af80.pub
apt-get update
apt-get install -qq cuda gcc-5 g++-5 -y
ln -s /usr/bin/gcc-5 /usr/local/cuda/bin/gcc
ln -s /usr/bin/g++-5 /usr/local/cuda/bin/g++
/usr/local/cuda/bin/nvcc --version
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment