environment.yml

name: gnetdgp_env
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - blas=1.0=mkl
  - ca-certificates=2020.12.5=h5b45459_0
  - certifi=2020.12.5=py38haa244fe_1
  - click=7.1.2=pyhd3eb1b0_0
  - cudatoolkit=10.2.89=h74a9793_1
  - intel-openmp=2021.2.0=haa95532_616
  - joblib=1.0.1=pyhd8ed1ab_0
  - libblas=3.9.0=1_h8933c1f_netlib
  - libcblas=3.9.0=5_hd5c7e75_netlib
  - liblapack=3.9.0=5_hd5c7e75_netlib
  - libuv=1.40.0=he774522_0
  - m2w64-gcc-libgfortran=5.3.0=6
  - m2w64-gcc-libs=5.3.0=7
  - m2w64-gcc-libs-core=5.3.0=7
  - m2w64-gmp=6.1.0=2
  - m2w64-libwinpthread-git=5.0.0.4634.697f757=2
  - mkl=2021.2.0=haa95532_296
  - mkl-service=2.3.0=py38h2bbff1b_1
  - mkl_fft=1.3.0=py38h277e83a_2
  - mkl_random=1.2.1=py38hf11a4ad_2
  - msys2-conda-epoch=20160418=1
  - ninja=1.10.2=h6d14046_1
  - numpy=1.20.1=py38h34a8a5c_0
  - numpy-base=1.20.1=py38haf7ebc8_0
  - openssl=1.1.1k=h8ffe710_0
  - pip=21.0.1=py38haa95532_0
  - python=3.8.8=hdbf39b2_5
  - python_abi=3.8=1_cp38
  - pytorch=1.4.0py3.8_cuda10.2_cudnn7_0
  - scikit-learn=0.24.2=py38h5d5d464_0
  - scipy=1.6.3=py38he847743_0
  - setuptools=52.0.0=py38haa95532_0
  - six=1.15.0=py38haa95532_0
  - sqlite=3.35.4=h2bbff1b_0
  - threadpoolctl=2.1.0=pyh5ca1d4c_0
  - typing_extensions=3.7.4.3=pyha847dfd_0
  - vc=14.2=h21ff451_1
  - vs2015_runtime=14.27.29016=h5e58377_2
  - wheel=0.36.2=pyhd3eb1b0_0
  - wincertstore=0.2=py38_0
  - pip:
    - --find-links https://pytorch-geometric.com/whl/torch-1.4.0+cpu.html
    - --requirement torch-scatter==2.0.4 torch-sparse==0.6.0 torch-cluster==1.5.4 torch-spline-conv==1.2.0
  - pip:
      - --requirement tqdm==4.46.1 optuna==1.5.0 colorama==0.4.3 colorlog==4.1.0 pandas==1.0.5
prefix: /home/user/conda/envs/gnetdgp_env