35 lines
920 B
Plaintext
35 lines
920 B
Plaintext
gymnasium<1.0.0
|
|
jax-jumpy==1.0.0 ; python_version >= "3.8" and python_version < "3.11"
|
|
matplotlib
|
|
moviepy
|
|
numpy<2.0
|
|
pandas
|
|
protobuf
|
|
pygame
|
|
stable-baselines3
|
|
tqdm
|
|
wandb
|
|
torchrl==0.7.2
|
|
tensordict==0.7.2
|
|
tyro
|
|
loguru
|
|
git+https://github.com/younggyoseo/mujoco_playground.git
|
|
torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
|
|
torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
|
|
torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
|
|
jax[cuda12]==0.4.35
|
|
nvidia-cublas-cu12==12.4.5.8
|
|
nvidia-cuda-cupti-cu12==12.4.127
|
|
nvidia-cuda-nvcc-cu12==12.8.93
|
|
nvidia-cuda-nvrtc-cu12==12.4.127
|
|
nvidia-cuda-runtime-cu12==12.4.127
|
|
nvidia-cudnn-cu12==9.1.0.70
|
|
nvidia-cufft-cu12==11.2.1.3
|
|
nvidia-curand-cu12==10.3.5.147
|
|
nvidia-cusolver-cu12==11.6.1.9
|
|
nvidia-cusparse-cu12==12.3.1.170
|
|
nvidia-cusparselt-cu12==0.6.2
|
|
nvidia-nccl-cu12==2.21.5
|
|
nvidia-nvjitlink-cu12==12.4.127
|
|
nvidia-nvtx-cu12==12.4.127
|