From 3485c78c6470ddb458b9c028b8e3f80850c729d7 Mon Sep 17 00:00:00 2001
From: alessiamarcolini <98marcolini@gmail.com>
Date: Wed, 15 Apr 2020 16:50:06 +0200
Subject: [PATCH] Fix environment file

---
 README.md |  2 +-
 env.yml   | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)
 create mode 100644 env.yml

diff --git a/README.md b/README.md
index 98018a3..592e7eb 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@
 ## Installation
 
 ```
-$ conda create --name simclr python=3.7 --file requirements.txt
+$ conda env create --name simclr --file env.yml
 $ conda activate simclr
 $ python run.py
 ```
diff --git a/env.yml b/env.yml
new file mode 100644
index 0000000..bb6cde3
--- /dev/null
+++ b/env.yml
@@ -0,0 +1,21 @@
+name: simclr
+channels:
+  - pytorch
+  - anaconda
+  - conda-forge
+  - defaults
+dependencies:
+  - cudatoolkit=10.1
+  - numpy=1.18.1
+  - opencv=3.4.2
+  - pillow=7.0
+  - pip=20.0
+  - python=3.7.6
+  - pytorch=1.4.0
+  - torchvision=0.5
+  - tensorboard=2.1
+  - matplotlib
+  - scikit-learn
+  - pyyaml=5.3.1
+  - nvidia-apex=0.1
+