-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements_gpu.txt
128 lines (128 loc) · 2.51 KB
/
requirements_gpu.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
absl-py==2.1.0
appdirs==1.4.4
asttokens==2.4.1
av==11.0.0
blinker==1.7.0
brax==0.10.0
certifi==2024.2.2
cfgv==3.4.0
charset-normalizer==3.3.2
chex==0.1.85
click==8.1.7
cloudpickle==3.0.0
comm==0.2.1
commit-linter==1.0.3
contextlib2==21.6.0
contourpy==1.2.0
cycler==0.12.1
debugpy==1.8.1
decorator==5.1.1
distlib==0.3.8
distrax==0.1.5
dm-env==1.6
dm-tree==0.1.8
docker-pycreds==0.4.0
etils==1.5.2
evosax==0.1.5
exceptiongroup==1.2.0
executing==2.0.1
filelock==3.13.1
Flask==3.0.2
Flask-Cors==4.0.0
flax==0.8.1
fonttools==4.49.0
fsspec==2024.2.0
gast==0.5.4
gitdb==4.0.11
GitPython==3.1.42
glfw==2.6.5
grpcio==1.62.0
gym==0.26.2
gym-notices==0.0.8
gymnax @ git+https://github.com/Ziksby/gymnax.git@d6a2117736dcdfe743878390a66b7977623bd962
identify==2.5.35
idna==3.6
imageio==2.34.0
imageio-ffmpeg==0.4.9
importlib-metadata==7.0.1
importlib-resources==6.1.1
ipykernel==6.29.2
ipython==8.18.1
itsdangerous==2.1.2
jax==0.4.25
jaxlib @ https://storage.googleapis.com/jax-releases/cuda12/jaxlib-0.4.23+cuda12.cudnn89-cp310-cp310-manylinux2014_x86_64.whl#sha256=8e42000672599e7ec0ea7f551acfcc95dcdd0e22b05a1d1f12f97b56a9fce4a8
jaxopt==0.8.3
jedi==0.19.1
Jinja2==3.1.3
jupyter_client==8.6.0
jupyter_core==5.7.1
kiwisolver==1.4.5
markdown-it-py==3.0.0
MarkupSafe==2.1.5
matplotlib==3.8.3
matplotlib-inline==0.1.6
mdurl==0.1.2
ml-dtypes==0.3.2
ml_collections==0.1.1
msgpack==1.0.7
mujoco==3.1.2
mujoco-mjx==3.1.2
nest-asyncio==1.6.0
nodeenv==1.8.0
numpy==1.26.4
nvidia-cublas-cu12==12.3.4.1
nvidia-cuda-cupti-cu12==12.3.101
nvidia-cuda-nvcc-cu12==12.3.107
nvidia-cuda-nvrtc-cu12==12.3.107
nvidia-cuda-runtime-cu12==12.3.101
nvidia-cudnn-cu12==8.9.7.29
nvidia-cufft-cu12==11.0.12.1
nvidia-cusolver-cu12==11.5.4.101
nvidia-cusparse-cu12==12.2.0.103
nvidia-nccl-cu12==2.19.3
nvidia-nvjitlink-cu12==12.3.101
opt-einsum==3.3.0
optax==0.1.9
orbax-checkpoint==0.5.3
packaging==23.2
parso==0.8.3
pexpect==4.9.0
pillow==10.2.0
platformdirs==4.2.0
pre-commit==3.4.0
prompt-toolkit==3.0.43
protobuf==4.25.3
psutil==5.9.8
ptyprocess==0.7.0
pure-eval==0.2.2
Pygments==2.17.2
PyOpenGL==3.1.7
pyparsing==3.1.1
python-dateutil==2.8.2
pytinyrenderer==0.0.14
PyYAML==6.0.1
pyzmq==25.1.2
requests==2.31.0
rich==13.7.0
scipy==1.12.0
sentry-sdk==1.40.6
setproctitle==1.3.3
six==1.16.0
smmap==5.0.1
stack-data==0.6.3
tensorboardX==2.6.2.2
tensorflow-probability==0.23.0
tensorstore==0.1.53
toolz==0.12.1
tornado==6.4
tqdm==4.66.2
traitlets==5.14.1
trimesh==4.1.4
typing_extensions==4.9.0
urllib3==2.2.1
virtualenv==20.25.1
wandb==0.16.3
wcwidth==0.2.13
Werkzeug==3.0.1
xminigrid==0.6.0
zipp==3.17.0