setup.sh

#!/bin/bash
set -e

module purge
module load cuda11.8/toolkit/11.8.0

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
bash ~/miniconda.sh -b -p $HOME/miniconda3
rm ~/miniconda.sh

source ~/miniconda3/etc/profile.d/conda.sh

conda create -n "vizdoom" python="$3.11" pip -y
conda activate "$vizdoom"

pip install --upgrade pip

pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install --no-cache-dir torchrl vizdoom pettingzoo opencv-python 'wandb[media]' tensorboard pyyaml cmake
pip install --no-cache-dir git+https://github.com/facebookresearch/BenchMARL.git