Added scikit-learn to the env

imedslab · Aug 31, 2018 · 04bd38c · 04bd38c
1 parent 57b0168
commit 04bd38c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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