Emilio Nicoli randomtask2000
randomtask2000
/ pytorch_setup.sh
Created
January 17, 2020 12:52
— forked from kylemcdonald/pytorch_setup.sh
Install CUDA 9.2, cuDNN 7.2.1, Anaconda and PyTorch on Ubuntu 16.04.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # tested on AWS p2.xlarge August 29, 2018 | |
| # install CUDA | |
| sudo apt-get update && sudo apt-get install wget -y --no-install-recommends | |
| CUDA_URL="https://developer.nvidia.com/compute/cuda/9.2/Prod2/local_installers/cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64" | |
| wget -c ${CUDA_URL} -O cuda.deb | |
| sudo dpkg --install cuda.deb | |
| sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub | |
| sudo apt-get update | |
| sudo apt-get install -y cuda |