Pārlūkot izejas kodu

edit requirments to enable cuda

Jhjoon05 2 gadi atpakaļ
vecāks
revīzija
f5c0c95a7b
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. 1 0
      requirements.txt

+ 1 - 0
requirements.txt

@@ -1,4 +1,5 @@
 numpy
+--extra-index-url https://download.pytorch.org/whl/cu117
 torch
 tqdm
 more-itertools