-
Notifications
You must be signed in to change notification settings - Fork 3
/
jax_tpu.yml
117 lines (117 loc) · 2.78 KB
/
jax_tpu.yml
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
name: jax_tpu
channels:
- defaults
dependencies:
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- bzip2=1.0.8=h7b6447c_0
- ca-certificates=2022.07.19=h06a4308_0
- certifi=2022.6.15=py310h06a4308_0
- ld_impl_linux-64=2.38=h1181459_1
- libffi=3.3=he6710b0_2
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libstdcxx-ng=11.2.0=h1234567_1
- libuuid=1.0.3=h7f8727e_2
- ncurses=6.3=h5eee18b_3
- openssl=1.1.1q=h7f8727e_0
- python=3.10.4=h12debd9_0
- readline=8.1.2=h7f8727e_1
- setuptools=61.2.0=py310h06a4308_0
- sqlite=3.38.5=hc218d9a_0
- tk=8.6.12=h1ccaba5_0
- tzdata=2022a=hda174b7_0
- wheel=0.37.1=pyhd3eb1b0_0
- xz=5.2.5=h7f8727e_1
- zlib=1.2.12=h7f8727e_2
- pip:
- absl-py==1.2.0
- astunparse==1.6.3
- cachetools==5.2.0
- charset-normalizer==2.1.0
- chex==0.1.3
- click==8.1.3
- cloudpickle==2.1.0
- colorama==0.4.5
- commonmark==0.9.1
- cycler==0.11.0
- decorator==5.1.1
- distrax==0.1.2
- dm-tree==0.1.7
- docker-pycreds==0.4.0
- docopt==0.6.2
- etils==0.6.0
- flatbuffers==1.12
- flax==0.5.3
- fonttools==4.34.4
- gast==0.4.0
- gitdb==4.0.9
- gitpython==3.1.27
- google-auth==2.9.1
- google-auth-oauthlib==0.4.6
- google-pasta==0.2.0
- grpcio==1.47.0
- gym==0.25.1
- gym-notices==0.0.7
- h5py==3.7.0
- idna==3.3
- imageio==2.21.0
- importlib-resources==5.9.0
- jax==0.3.15
- jaxlib==0.3.15
- keras==2.9.0
- keras-preprocessing==1.1.2
- kiwisolver==1.4.4
- libclang==14.0.6
- markdown==3.4.1
- markupsafe==2.1.1
- matplotlib==3.5.2
- msgpack==1.0.4
- numpy==1.23.1
- oauthlib==3.2.0
- opencv-python==4.6.0.66
- opt-einsum==3.3.0
- optax==0.1.3
- packaging==21.3
- pathtools==0.1.2
- pillow==9.2.0
- pip==22.2.1
- pipreqs==0.4.11
- promise==2.3
- protobuf==3.19.4
- psutil==5.9.1
- pyasn1==0.4.8
- pyasn1-modules==0.2.8
- pygments==2.12.0
- pyparsing==3.0.9
- python-dateutil==2.8.2
- pyyaml==6.0
- requests==2.28.1
- requests-oauthlib==1.3.1
- rich==11.2.0
- rsa==4.9
- scipy==1.9.0
- sentry-sdk==1.9.0
- setproctitle==1.3.0
- shortuuid==1.0.9
- six==1.16.0
- smmap==5.0.0
- tensorboard==2.9.1
- tensorboard-data-server==0.6.1
- tensorboard-plugin-wit==1.8.1
- tensorflow==2.9.1
- tensorflow-estimator==2.9.0
- tensorflow-io-gcs-filesystem==0.26.0
- tensorflow-probability==0.17.0
- tensorstore==0.1.21
- termcolor==1.1.0
- toolz==0.12.0
- tqdm==4.64.0
- typing-extensions==4.3.0
- urllib3==1.26.11
- wandb==0.12.21
- werkzeug==2.2.1
- wrapt==1.14.1
- yarg==0.1.9
- zipp==3.8.1
prefix: /home/aviralkumar/miniconda3/envs/jax_tpu