vm_setup.sh

sudo apt-get install linux-headers-$(uname -r)
sudo apt-get update
sudo apt-get upgrade 
sudo apt-get install gcc
sudo apt-get install libxml2
sudo apt-get install make 
sudo apt-get install python3-pip
sudo apt-get install git
sudo apt-get install tmux

gcc --version

wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run

sudo sh cuda_11.7.1_515.65.01_linux.run


wget https://www.dropbox.com/s/ds5l73rlbmpamj9/cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz?dl=0

mv 'cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz?dl=0' cudnn8.5.tar.xz

tar xvf cudnn8*


sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

git clone https://github.com/sourabh2k15/algorithmic-efficiency.git
cd algorithmic-efficiency

pip3 install -e .
pip3 install -e '.[dev]'
pip3 install -e '.[pytorch_gpu]' -f 'https://download.pytorch.org/whl/torch_stable.html'
pip3 install -e '.[jax_gpu]' -f 'https://storage.googleapis.com/jax-releases/jax_cuda_releases.html'

cd ..

git clone https://github.com/coder/code-server.git                                
cd code-server
./install.sh