Setup CUDA 11.8 drivers, miniconda and pytorch on Ubuntu 22.04 on AWS EC2

aws-ec2-cuda118-ubuntu2204.sh

#!/bin/bash

apt update
apt -y install build-essential  # install GCC
apt -y install --reinstall software-properties-common # enable add-apt-repository
apt-get -y install lshw
apt-get clean

export DISTRO_UBUNTU=ubuntu2204/x86_64

# # Ubuntu 22 - : install CUDA 11.8 (for Pytorch support)
wget https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO_UBUNTU}/cuda-ubuntu2204.pin
mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
dpkg -i cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
cp /var/cuda-repo-ubuntu2204-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
apt-get update
apt-get -y install cuda

# ensure cuda binaries are on path
export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}

nvcc --version


curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh -b -f

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118