-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yml
103 lines (103 loc) · 2.81 KB
/
environment.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
name: hessavg
channels:
- defaults
dependencies:
- python=3.9.16
- pip
- _libgcc_mutex=0.1=main
- _openmp_mutex=5.1=1_gnu
- ca-certificates=2023.12.12=h06a4308_0
- ld_impl_linux-64=2.38=h1181459_1
- libffi=3.4.4=h6a678d5_0
- libgcc-ng=11.2.0=h1234567_1
- libgomp=11.2.0=h1234567_1
- libstdcxx-ng=11.2.0=h1234567_1
- ncurses=6.4=h6a678d5_0
- openssl=3.0.12=h7f8727e_0
- python=3.9.16=h955ad1f_3
- readline=8.2=h5eee18b_0
- setuptools=68.2.2=py39h06a4308_0
- sqlite=3.41.2=h5eee18b_0
- tk=8.6.12=h1ccaba5_0
- tzdata=2023d=h04d1e81_0
- wheel=0.41.2=py39h06a4308_0
- xz=5.4.5=h5eee18b_0
- zlib=1.2.13=h5eee18b_0
- pip:
- absl-py==2.0.0
- certifi==2023.11.17
- charset-normalizer==3.3.2
- chex==0.1.85
- contourpy==1.2.0
- cycler==0.12.1
- etils==1.5.2
- filelock==3.13.1
- flax==0.6.9
- fonttools==4.47.0
- fsspec==2023.12.2
- idna==3.6
- importlib-metadata==7.0.1
- importlib-resources==6.1.1
- jax==0.4.23
- jaxlib==0.4.23+cuda11.cudnn86
- jinja2==3.1.2
- joblib==1.3.2
- kiwisolver==1.4.5
- markdown-it-py==3.0.0
- markupsafe==2.1.3
- matplotlib==3.8.2
- mdurl==0.1.2
- ml-dtypes==0.3.2
- mpmath==1.3.0
- msgpack==1.0.7
- nest-asyncio==1.5.8
- networkx==3.2.1
- numpy==1.26.3
- nvidia-cublas-cu11==11.11.3.6
- nvidia-cublas-cu12==12.1.3.1
- nvidia-cuda-cupti-cu11==11.8.87
- nvidia-cuda-cupti-cu12==12.1.105
- nvidia-cuda-nvcc-cu11==11.8.89
- nvidia-cuda-nvrtc-cu11==11.8.89
- nvidia-cuda-nvrtc-cu12==12.1.105
- nvidia-cuda-runtime-cu11==11.8.89
- nvidia-cuda-runtime-cu12==12.1.105
- nvidia-cudnn-cu11==8.9.6.50
- nvidia-cudnn-cu12==8.9.2.26
- nvidia-cufft-cu11==10.9.0.58
- nvidia-cufft-cu12==11.0.2.54
- nvidia-curand-cu12==10.3.2.106
- nvidia-cusolver-cu11==11.4.1.48
- nvidia-cusolver-cu12==11.4.5.107
- nvidia-cusparse-cu11==11.7.5.86
- nvidia-cusparse-cu12==12.1.0.106
- nvidia-nccl-cu11==2.19.3
- nvidia-nccl-cu12==2.18.1
- nvidia-nvjitlink-cu12==12.3.101
- nvidia-nvtx-cu12==12.1.105
- opt-einsum==3.3.0
- optax==0.1.7
- orbax-checkpoint==0.4.8
- packaging==23.2
- pillow==10.2.0
- pip==23.3.2
- protobuf==4.25.1
- pygments==2.17.2
- pyparsing==3.1.1
- python-dateutil==2.8.2
- pyyaml==6.0.1
- requests==2.31.0
- rich==13.7.0
- scikit-learn==1.4.0
- scipy==1.11.4
- six==1.16.0
- sympy==1.12
- tensorstore==0.1.51
- threadpoolctl==3.2.0
- toolz==0.12.0
- torch==2.1.2
- torchvision==0.16.2
- triton==2.1.0
- typing-extensions==4.9.0
- urllib3==2.1.0
- zipp==3.17.0