shahriar1489
/ pytorch_setup.sh
Created
September 24, 2019 19:09
— forked from kylemcdonald/pytorch_setup.sh
Install CUDA 9.2, cuDNN 7.2.1, Anaconda and PyTorch on Ubuntu 16.04.
This file contains 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 |