Update setup.py to torch2

Vermeille · Sep 4, 2024 · 5c6c3da · 5c6c3da
1 parent 9311a9c
commit 5c6c3da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
           'visdom>=0.1.8',
           'crayons>=0.2',
           'torchvision>=0.13',
-          'torch>=1.12',
+          'torch>=2',
           'numpy>=1.16',
           'Pillow>=6',
       ])