From 2fbd6f876c34bfdbcbddc71cd73646e71bde5748 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 25 Mar 2021 01:18:04 -0500
Subject: [PATCH] Don't require strict PyTorch and TorchVision versions (hpvm
requirement)
---
setup.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/setup.py b/setup.py
index 3377b9c..2003818 100644
--- a/setup.py
+++ b/setup.py
@@ -16,8 +16,8 @@ setuptools.setup(
install_requires=[
"matplotlib>=3.3",
"networkx>=2.5",
- "torch==1.7.1",
- "torchvision==0.8.2",
+ "torch>=1.5.1",
+ "torchvision>=0.6",
"tqdm>=4.50",
"pandas>=1.1",
"jsonpickle>=1.5",
--
GitLab