diff --git a/env.yaml b/env.yaml
index b44c726..f544396 100644
--- a/env.yaml
+++ b/env.yaml
@@ -1,128 +1,109 @@
 name: PointFlow
+
 channels:
+  - nodefaults
+  - drti
   - pytorch
-  - defaults
+  - nvidia
+  - conda-forge
+  - pyg
+
 dependencies:
-  - _libgcc_mutex=0.1=main
-  - _openmp_mutex=4.5=1_gnu
-  - blas=1.0=mkl
-  - brotli=1.0.9=he6710b0_2
-  - brotlipy=0.7.0=py37h27cfd23_1003
-  - bzip2=1.0.8=h7b6447c_0
-  - ca-certificates=2022.3.29=h06a4308_0
-  - certifi=2021.10.8=py37h06a4308_2
-  - cffi=1.15.0=py37hd667e15_1
-  - charset-normalizer=2.0.4=pyhd3eb1b0_0
-  - colorama=0.4.4=pyhd3eb1b0_0
-  - cryptography=36.0.0=py37h9ce1e76_0
-  - cudatoolkit=10.2.89=hfd86e86_1
-  - cycler=0.11.0=pyhd3eb1b0_0
-  - dbus=1.13.18=hb2f20db_0
-  - expat=2.4.4=h295c915_0
-  - ffmpeg=4.3=hf484d3e_0
-  - fontconfig=2.13.1=h6c09931_0
-  - fonttools=4.25.0=pyhd3eb1b0_0
-  - freetype=2.11.0=h70c0345_0
-  - giflib=5.2.1=h7b6447c_0
-  - glib=2.69.1=h4ff587b_1
-  - gmp=6.2.1=h2531618_2
-  - gnutls=3.6.15=he1e5248_0
-  - gst-plugins-base=1.14.0=h8213a91_2
-  - gstreamer=1.14.0=h28cd5cc_2
-  - icu=58.2=he6710b0_3
-  - idna=3.3=pyhd3eb1b0_0
-  - intel-openmp=2021.4.0=h06a4308_3561
-  - joblib=1.1.0=pyhd3eb1b0_0
-  - jpeg=9d=h7f8727e_0
-  - kiwisolver=1.3.2=py37h295c915_0
-  - lame=3.100=h7b6447c_0
-  - lcms2=2.12=h3be6417_0
-  - ld_impl_linux-64=2.35.1=h7274673_9
-  - libffi=3.3=he6710b0_2
-  - libgcc-ng=9.3.0=h5101ec6_17
-  - libgfortran-ng=7.5.0=ha8ba4b0_17
-  - libgfortran4=7.5.0=ha8ba4b0_17
-  - libgomp=9.3.0=h5101ec6_17
-  - libiconv=1.15=h63c8f33_5
-  - libidn2=2.3.2=h7f8727e_0
-  - libpng=1.6.37=hbc83047_0
-  - libstdcxx-ng=9.3.0=hd4cf53a_17
-  - libtasn1=4.16.0=h27cfd23_0
-  - libtiff=4.2.0=h85742a9_0
-  - libunistring=0.9.10=h27cfd23_0
-  - libuuid=1.0.3=h7f8727e_2
-  - libuv=1.40.0=h7b6447c_0
-  - libwebp=1.2.2=h55f646e_0
-  - libwebp-base=1.2.2=h7f8727e_0
-  - libxcb=1.14=h7b6447c_0
-  - libxml2=2.9.12=h03d6c58_0
-  - lz4-c=1.9.3=h295c915_1
-  - matplotlib=3.5.1=py37h06a4308_1
-  - matplotlib-base=3.5.1=py37ha18d171_1
-  - mkl=2021.4.0=h06a4308_640
-  - mkl-service=2.4.0=py37h7f8727e_0
-  - mkl_fft=1.3.1=py37hd3c417c_0
-  - mkl_random=1.2.2=py37h51133e4_0
-  - munkres=1.1.4=py_0
-  - ncurses=6.3=h7f8727e_2
-  - nettle=3.7.3=hbbd107a_1
-  - numpy-base=1.21.2=py37h79a1101_0
-  - openh264=2.1.1=h4ff587b_0
-  - openssl=1.1.1n=h7f8727e_0
-  - packaging=21.3=pyhd3eb1b0_0
-  - pcre=8.45=h295c915_0
-  - pip=21.2.2=py37h06a4308_0
-  - pycparser=2.21=pyhd3eb1b0_0
-  - pyopenssl=22.0.0=pyhd3eb1b0_0
-  - pyparsing=3.0.4=pyhd3eb1b0_0
-  - pyqt=5.9.2=py37h05f1152_2
-  - pysocks=1.7.1=py37_1
-  - python=3.7.13=h12debd9_0
-  - python-dateutil=2.8.2=pyhd3eb1b0_0
-  - pytorch=1.11.0=py3.7_cuda10.2_cudnn7.6.5_0
-  - pytorch-mutex=1.0=cuda
-  - qt=5.9.7=h5867ecd_1
-  - readline=8.1.2=h7f8727e_1
-  - requests=2.27.1=pyhd3eb1b0_0
-  - scikit-learn=1.0.2=py37h51133e4_1
-  - scipy=1.7.3=py37hc147768_0
-  - setuptools=58.0.4=py37h06a4308_0
-  - sip=4.19.8=py37hf484d3e_0
-  - six=1.16.0=pyhd3eb1b0_1
-  - sqlite=3.38.2=hc218d9a_0
-  - threadpoolctl=2.2.0=pyh0d69192_0
-  - tk=8.6.11=h1ccaba5_0
-  - torchaudio=0.11.0=py37_cu102
-  - torchvision=0.12.0=py37_cu102
-  - tornado=6.1=py37h27cfd23_0
-  - tqdm=4.63.0=pyhd3eb1b0_0
-  - typing_extensions=4.1.1=pyh06a4308_0
-  - urllib3=1.26.8=pyhd3eb1b0_0
-  - wheel=0.37.1=pyhd3eb1b0_0
-  - xz=5.2.5=h7b6447c_0
-  - zlib=1.2.11=h7f8727e_4
-  - zstd=1.4.9=haebb681_0
-  - pip:
-    - absl-py==1.0.0
-    - astor==0.8.1
-    - cached-property==1.5.2
-    - gast==0.5.3
-    - grpcio==1.44.0
-    - h5py==3.6.0
-    - importlib-metadata==4.11.3
-    - keras-applications==1.0.8
-    - keras-preprocessing==1.1.2
-    - markdown==3.3.6
-    - mock==4.0.3
-    - numpy==1.16.4
-    - pillow==5.0.0
-    - protobuf==3.20.0
-    - tensorboard==1.13.1
-    - tensorboardx==1.7
-    - tensorflow-estimator==1.13.0
-    - tensorflow-gpu==1.13.1
-    - termcolor==1.1.0
-    - torchdiffeq==0.0.1
-    - werkzeug==2.1.1
-    - zipp==3.8.0
-prefix: /home/grendelyang/anaconda3/envs/PointFlow
+  - _libgcc_mutex
+  - _openmp_mutex
+  - blas
+  - brotli
+  - brotlipy
+  - bzip2
+  - ca-certificates
+  - certifi
+  - cffi
+  - charset-normalizer
+  - colorama
+  - cryptography
+  - cudatoolkit
+  - cycler
+  - dbus
+  - expat
+  - ffmpeg
+  - fontconfig
+  - fonttools
+  - freetype
+  - giflib
+  - glib
+  - gmp
+  - gnutls
+  - gst-plugins-base
+  - gstreamer
+  - icu
+  - idna
+  - joblib
+  - jpeg
+  - kiwisolver
+  - lame
+  - lcms2
+  - ld_impl_linux-64
+  - libffi
+  - libgcc-ng
+  - libgfortran-ng
+  - libgfortran4
+  - libgomp
+  - libiconv
+  - libidn2
+  - libpng
+  - libstdcxx-ng
+  - libtasn1
+  - libtiff
+  - libunistring
+  - libuuid
+  - libuv
+  - libwebp
+  - libwebp-base
+  - libxcb
+  - libxml2
+  - lz4-c
+  - matplotlib
+  - matplotlib-base
+  - mkl
+  - mkl-service
+  - mkl_fft
+  - mkl_random
+  - munkres
+  - ncurses
+  - nettle
+  - openh264
+  - openssl
+  - packaging
+  - pcre
+  - pip
+  - pycparser
+  - pyopenssl
+  - pyparsing
+  - pyqt
+  - pysocks
+  - python
+  - python-dateutil
+  - pytorch
+  - pytorch-mutex
+  - qt
+  - readline
+  - requests
+  - scikit-learn
+  - scipy
+  - setuptools
+  - sip
+  - six
+  - sqlite
+  - threadpoolctl
+  - tk
+  - torchaudio
+  - torchvision
+  - tornado
+  - tqdm
+  - typing_extensions
+  - urllib3
+  - wheel
+  - xz
+  - zlib
+  - zstd
+  - torchdiffeq
\ No newline at end of file