2021-05-17 18:24:13 +02:00
|
|
|
# Deep learning.
|
|
|
|
|
# --------------
|
|
|
|
|
tensorflow==2.4.1
|
|
|
|
|
tensorflow-probability==0.12.2
|
|
|
|
|
torch==1.8.1;sys_platform=="darwin"
|
|
|
|
|
torchvision==0.9.1;sys_platform=="darwin"
|
|
|
|
|
|
|
|
|
|
# On non-OSX machines only install CPU version of torch and torchvision
|
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
|
torch==1.8.1+cpu;sys_platform!="darwin"
|
|
|
|
|
-f https://download.pytorch.org/whl/torch_stable.html
|
|
|
|
|
torchvision==0.9.1+cpu;sys_platform!="darwin"
|
|
|
|
|
|
|
|
|
|
# Environment adapters.
|
|
|
|
|
# ---------------------
|
|
|
|
|
# Atari
|
|
|
|
|
atari_py==0.2.6
|
2021-07-02 09:58:18 -07:00
|
|
|
gym[atari]==0.18.3
|
2021-05-17 18:24:13 +02:00
|
|
|
# Kaggle envs.
|
|
|
|
|
kaggle_environments==1.7.11
|
|
|
|
|
# Unity3D testing
|
|
|
|
|
mlagents==0.26.0
|
|
|
|
|
mlagents_envs==0.26.0
|
|
|
|
|
# For tests on PettingZoo's multi-agent envs.
|
2021-05-18 11:25:52 +02:00
|
|
|
pettingzoo==1.8.2
|
2021-07-13 12:38:14 +02:00
|
|
|
pymunk
|
|
|
|
|
supersuit
|
2021-05-17 18:24:13 +02:00
|
|
|
# For testing in MuJoCo-like envs (in PyBullet).
|
|
|
|
|
pybullet==3.1.7
|
|
|
|
|
# For tests on RecSim and Kaggle envs.
|
|
|
|
|
recsim==0.2.4
|
|
|
|
|
|
|
|
|
|
# Other.
|
|
|
|
|
# ------
|
|
|
|
|
# For MAML on PyTorch.
|
|
|
|
|
higher==0.2.1
|
|
|
|
|
# For auto-generating an env-rendering Window.
|
|
|
|
|
pyglet==1.5.0
|
|
|
|
|
# For JSON reader/writer.
|
|
|
|
|
smart_open==5.0.0
|
|
|
|
|
# Ray Serve example
|
|
|
|
|
starlette==0.14.2
|
