team222 / requirements.txt
ylop's picture
Update CUDA to 12.1 for L40S GPU support
16dce8f verified
raw
history blame
251 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch>=2.0.0
gymnasium>=0.29.0
pygame>=2.5.0
numpy>=1.24.0
stable-baselines3>=2.0.0
matplotlib>=3.7.0
tensorboard>=2.14.0
tqdm>=4.65.0
rich>=13.0.0
PyFlyt>=0.3.0
opensimplex>=0.4.4
ncps>=0.0.7