diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..072fc1f
--- /dev/null
+++ b/environment.yml
@@ -0,0 +1,84 @@
+name: swapping_AE
+channels:
+  - pytorch
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=4.5=1_gnu
+  - blas=1.0=mkl
+  - ca-certificates=2021.10.26=h06a4308_2
+  - certifi=2021.5.30=py36h06a4308_0
+  - cudatoolkit=11.0.221=h6bb024c_0
+  - dataclasses=0.8=pyh4f3eec9_6
+  - freetype=2.11.0=h70c0345_0
+  - intel-openmp=2022.0.1=h06a4308_3633
+  - jpeg=9b=h024ee3a_2
+  - 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
+  - libgomp=9.3.0=h5101ec6_17
+  - libpng=1.6.37=hbc83047_0
+  - libstdcxx-ng=9.3.0=hd4cf53a_17
+  - libtiff=4.2.0=h85742a9_0
+  - libuv=1.40.0=h7b6447c_0
+  - libwebp-base=1.2.0=h27cfd23_0
+  - lz4-c=1.9.3=h295c915_1
+  - mkl=2020.2=256
+  - mkl-service=2.3.0=py36he8ac12f_0
+  - mkl_fft=1.3.0=py36h54f3939_0
+  - mkl_random=1.1.1=py36h0573a6f_0
+  - ncurses=6.3=h7f8727e_2
+  - numpy-base=1.19.2=py36hfa32c7d_0
+  - olefile=0.46=py36_0
+  - openjpeg=2.4.0=h3ad879b_0
+  - openssl=1.1.1m=h7f8727e_0
+  - pip=21.2.2=py36h06a4308_0
+  - python=3.6.13=h12debd9_1
+  - pytorch=1.7.1=py3.6_cuda11.0.221_cudnn8.0.5_0
+  - readline=8.1.2=h7f8727e_1
+  - setuptools=58.0.4=py36h06a4308_0
+  - six=1.16.0=pyhd3eb1b0_0
+  - sqlite=3.37.0=hc218d9a_0
+  - tk=8.6.11=h1ccaba5_0
+  - typing_extensions=3.10.0.2=pyh06a4308_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:
+    - charset-normalizer==2.0.10
+    - cycler==0.11.0
+    - dominate==2.6.0
+    - func-timeout==4.3.5
+    - gputil==1.4.0
+    - idna==3.3
+    - joblib==1.1.0
+    - jsonpatch==1.32
+    - jsonpointer==2.2
+    - kiwisolver==1.3.1
+    - lmdb==1.3.0
+    - matplotlib==3.3.4
+    - ninja==1.10.2.3
+    - numpy==1.19.5
+    - opencv-python==4.5.5.62
+    - pillow==8.4.0
+    - pyparsing==3.0.7
+    - python-dateutil==2.8.2
+    - pyzmq==22.3.0
+    - requests==2.27.1
+    - scikit-learn==0.24.2
+    - scipy==1.5.4
+    - threadpoolctl==3.0.0
+    - torch==1.7.1+cu110
+    - torchaudio==0.7.2
+    - torchfile==0.1.0
+    - torchgeometry==0.1.2
+    - torchvision==0.8.2+cu110
+    - tornado==6.1
+    - tqdm==4.62.3
+    - typing-extensions==4.0.1
+    - urllib3==1.26.8
+    - visdom==0.1.8.9
+    - websocket-client==1.2.3
+prefix: /home/ece/anaconda3/envs/swapping_AE