diff --git a/create_conda_env.sh b/create_conda_env.sh
index c60fb55..6cd5abb 100644
--- a/create_conda_env.sh
+++ b/create_conda_env.sh
@@ -3,8 +3,8 @@
 conda create -y -n deep_knee python=3.6
 conda activate deep_knee
 
-conda install -y numpy opencv scipy pyyaml cython matplotlib
-conda install -y pytorch=0.3.1 -c soumith
+conda install -y numpy opencv scipy pyyaml cython matplotlib scikit-learn
+conda install -y pytorch==0.3.1 -c pytorch
 conda install -y git-lfs -c conda-forge
 
 pip install pip -U