Merge pull request IrisRainbowNeko#1 from FishMagic/master

Fix version flag error
EndedEx · Sep 21, 2021 · cb7d23f · cb7d23f
2 parents 87a7f7d + b36027c
commit cb7d23f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/requirements.py b/requirements.py
@@ -20,11 +20,11 @@
 Pillow
 pymouse
 numpy==1.19.5
-torch==1.7.0+{"cpu" if args.cuda is None else args.cuda} -f https://download.pytorch.org/whl/torch_stable.html
---no-deps torchvision==0.8.1+{"cpu" if args.cuda is None else args.cuda} -f https://download.pytorch.org/whl/torch_stable.html
+torch==1.7.0+{"cpu" if args.cuda is None else "cu" + args.cuda} -f https://download.pytorch.org/whl/torch_stable.html
+--no-deps torchvision==0.8.1+{"cpu" if args.cuda is None else "cu" + args.cuda} -f https://download.pytorch.org/whl/torch_stable.html
 thop
 git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
 '''
 
 for line in pkgs.split('\n'):
-    pip.main(['install', *line.split()])
+    pip.main(['install', *line.split()])