Setup
- conda create --name lm python=3.10
- Install Pytorch with conda via official instructions, e.g.
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
conda install scipy
pip install -r requirements.txt
- Install Atari-Py roms or maybe use just
pip install -U gym[atari]
pip install -U gym[atari]
if you didn't on the previous step. Also perhapsale-import-roms --import-from-pkg atari_py.atari_roms
or justale-import-roms
Checkpoints