-
Notifications
You must be signed in to change notification settings - Fork 1
/
environment.yml
54 lines (54 loc) · 1.57 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
name: MPRL
channels:
- defaults
- anaconda
- conda-forge
dependencies:
- _libgcc_mutex=0.1=conda_forge
- _openmp_mutex=4.5=2_gnu
- ca-certificates=2024.7.4=hbcca054_0
- cudatoolkit=11.2.2=hc23eb0c_13
- cudnn=8.1.0.77=h90431f1_0
- ld_impl_linux-64=2.38=h1181459_1
- libffi=3.4.4=h6a678d5_1
- libgcc-ng=14.1.0=h77fa898_0
- libgomp=14.1.0=h77fa898_0
- libstdcxx-ng=14.1.0=hc0a3c3a_0
- ncurses=6.4=h6a678d5_0
- openssl=3.3.1=h4bc722e_2
- pip=24.0=py38h06a4308_0
- python=3.8.19=h955ad1f_0
- readline=8.2=h5eee18b_0
- setuptools=69.5.1=py38h06a4308_0
- sqlite=3.45.3=h5eee18b_0
- tk=8.6.14=h39e8969_0
- wheel=0.43.0=py38h06a4308_0
- xz=5.4.6=h5eee18b_1
- zlib=1.2.13=h5eee18b_1
- pip:
- charset-normalizer==3.3.2
- idna==3.7
- nvidia-cublas-cu11==11.11.3.6
- nvidia-cuda-cupti-cu11==11.8.87
- nvidia-cuda-nvrtc-cu11==11.8.89
- nvidia-cuda-runtime-cu11==11.8.89
- nvidia-cudnn-cu11==8.7.0.84
- nvidia-cufft-cu11==10.9.0.58
- nvidia-curand-cu11==10.3.0.86
- nvidia-cusolver-cu11==11.4.1.48
- nvidia-cusparse-cu11==11.7.5.86
- nvidia-nccl-cu11==2.19.3
- nvidia-nccl-cu12==2.20.5
- nvidia-nvtx-cu11==11.8.86
- psutil==6.0.0
- pyg-lib==0.4.0+pt22cu118
- requests==2.32.3
- torch==2.2.2+cu118
- torch-cluster==1.6.3+pt22cu118
- torch-scatter==2.1.2+pt22cu118
- torch-sparse==0.6.18+pt22cu118
- torch-spline-conv==1.2.2+pt22cu118
- torchaudio==2.2.2+cu118
- torchvision==0.17.2+cu118
- triton==2.2.0
- xgboost==2.1.0