shahriar1489

🎯

-______________-

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.

	# 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