From 72294fef306095d071242d4482d4c0bd9b5a8dac Mon Sep 17 00:00:00 2001
From: Ivan Razumov <ivan.razumov@cern.ch>
Date: Thu, 5 Nov 2020 09:59:01 +0100
Subject: [PATCH] Update torch and torchvision to fix compatibility with python
 3.8

---
 cmake/toolchain/heptools-dev-base.cmake | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/cmake/toolchain/heptools-dev-base.cmake b/cmake/toolchain/heptools-dev-base.cmake
index c97d26802d..bb3469afb0 100644
--- a/cmake/toolchain/heptools-dev-base.cmake
+++ b/cmake/toolchain/heptools-dev-base.cmake
@@ -566,14 +566,8 @@ LCG_external_package(thrift            0.13.0
 LCG_external_package(tiff              4.0.10                                   )
 LCG_external_package(toolz             0.9.0                                    )
 if(NOT ${LCG_OS}${LCG_OSVERS} STREQUAL slc6)
-  if(((${LCG_COMP} MATCHES gcc) AND (${LCG_COMPVERS} LESS 9)) OR (${LCG_COMP} MATCHES clang))
-     LCG_external_package(torch             1.4.0                                    )
-     LCG_external_package(torchvision       0.5.0                                    )
-   else()
-     # Keep, for the moment, an old version for gcc9 (see https://github.com/pytorch/pytorch/issues/32277) (GG 2/3/2020)
-     LCG_external_package(torch             1.3.0                                    )
-     LCG_external_package(torchvision       0.4.1                                    )
-  endif()
+  LCG_external_package(torch             1.7.0                                    )
+  LCG_external_package(torchvision       0.8.1                                    )
 endif()
 LCG_external_package(tornado           5.1.1                                    )
 LCG_external_package(traitlets         4.3.2                                    )
-- 
GitLab