From 43470097675774589e25762b01697490783b8609 Mon Sep 17 00:00:00 2001
From: edgarriba <edgar.riba@gmail.com>
Date: Mon, 11 Feb 2019 14:40:28 +0100
Subject: [PATCH] update to pytorch v1.0.1

---
 setup_dev_env.sh    | 2 +-
 setup_travis_env.sh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup_dev_env.sh b/setup_dev_env.sh
index 5338541862..e5c437532b 100755
--- 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
index ffabc7f7dd..1908f79417 100755
--- 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