Skip to content

Instantly share code, notes, and snippets.

@yevshev

yevshev/nvidia.sh

Last active November 28, 2019 20:02
Show Gist options
  • Save yevshev/9a8601d4674c1a09d8c00b70ee6d8df1 to your computer and use it in GitHub Desktop.
Save yevshev/9a8601d4674c1a09d8c00b70ee6d8df1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nvidia.sh
nvkey=http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
nvlist=/etc/apt/sources.list.d/nvidia.list
cudapath=/usr/local/cuda
apt-key adv --fetch-keys $nvkey
echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > $nvlist
echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" >> $nvlist
apt update
apt install nvidia-driver-440 cuda-10-0 libcudnn7 libcudnn7-dev libnvinfer5 libnvinfer5-dev -y
export PATH=$PATH:$cudapath/bin
export LD_LIBRARY_PATH:$LD_LIBRARY_PATH:$cudapath/lib64
export LD_LIBRARY_PATH:$LD_LIBRARY_PATH:$cudapath/extras/CUPTI/lib64
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment