catid / crafty

Experimental solver for crafter

crafty

Leveraging foundation models to do well in the crafter gym.

Setup

Install CUDA toolkit:

# Here's how to do it on Ubuntu:
sudo apt install nvidia-cuda-toolkit

Make sure you can use nvcc:

nvcc --version

git clone https://github.com/catid/crafty
cd crafty

conda create -n crafty python=3.10
conda activate crafty

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

pip install -U -r requirements.txt

pip install ninja
pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers

pip install git+https://github.com/catid/crafter.git#egg=crafter

# Update this from https://github.com/NVIDIA/DALI#installing-dali
pip install --upgrade nvidia-dali-cuda110 --extra-index-url https://developer.download.nvidia.com/compute/redist

Train

python train_recon.py

Kill hanging processes:

pkill -f train_recon.py