Merge pull request #72 from arraiyopensource/fix/pytorch_v011

update to pytorch v1.0.1
kornia · Feb 12, 2019 · 5c9d9ae · 5c9d9ae
2 parents f24630c + 4347009
commit 5c9d9ae
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