mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
41 lines
1,001 B
Text
41 lines
1,001 B
Text
|
# 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
|
||
|
|
gym[atari]==0.18.0
|
||
|
|
# 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.
|
||
|
|
pettingzoo==1.8.1
|
||
|
|
# 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
|