From 9e4e0fc9abbbfccde37ca0170620911cbf47dcf8 Mon Sep 17 00:00:00 2001
From: MishaLaskin <laskin.misha@gmail.com>
Date: Thu, 20 Feb 2020 02:48:52 -0800
Subject: [PATCH] readme and installation

---
 README.md     | 8 ++++++++
 conda_env.yml | 5 ++++-
 utils.py      | 1 -
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 5aad476..aa92a18 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,13 @@
 # CURL: Contrastive Unsupervised Representation Learning for Sample-Efficient Reinforcement Learning
 
+## Installation 
+
+All of the dependencies are in the `conda_env.yml` file. They can be installed manually or with the following command:
+
+```
+conda env create -f conda_env.yml
+```
+
 ## Instructions
 To train a CURL agent on the `cartpole swingup` task from image-based observations run `bash script/run.sh` from the root of this directory. The `run.sh` file contains the following command, which you can modify to try different environments / hyperparamters.
 ```
diff --git a/conda_env.yml b/conda_env.yml
index df635fa..fffc1a5 100644
--- a/conda_env.yml
+++ b/conda_env.yml
@@ -8,10 +8,13 @@ dependencies:
   - cudatoolkit=9.2
   - absl-py
   - pyparsing
+  - pillow=6.1
   - pip:
     - termcolor
     - git+git://github.com/deepmind/dm_control.git
     - git+git://github.com/1nadequacy/dmc2gym.git
     - tb-nightly
     - imageio
-    - imageio-ffmpeg
\ No newline at end of file
+    - imageio-ffmpeg
+    - torchvision
+    - scikit-image
\ No newline at end of file
diff --git a/utils.py b/utils.py
index 76514e1..ac2e150 100644
--- a/utils.py
+++ b/utils.py
@@ -6,7 +6,6 @@ import os
 from collections import deque
 import random
 from torch.utils.data import Dataset, DataLoader
-from torchvision import transforms
 import time
 from skimage.util.shape import view_as_windows