update to pytorch v1.0.1

kornia · Feb 11, 2019 · 4347009 · 4347009 · versatran01 · Feb 13, 2019
1 parent f24630c
commit 4347009
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup_dev_env.sh b/setup_dev_env.sh
@@ -23,7 +23,7 @@ $sdk_dir/.dev_env/bin/conda install -y \
   jupyter \
   matplotlib \
   numpy \
-  pytorch \
+  pytorch==1.0.1 \
   torchvision \
   opencv \
   -c pytorch

diff --git a/setup_travis_env.sh b/setup_travis_env.sh
@@ -19,7 +19,7 @@ fi
 
 # Install CPU-PyTorch
 $sdk_dir/.dev_env/bin/conda install -y \
-  pytorch-cpu==1.0.0 \
+  pytorch-cpu==1.0.1 \
   -c pytorch
 
 # Tests dependencies