faster inference with cuda/cudnn backends flags

2022-07-09 06:48:51 -04:00
parent 703bfb231d
commit dba3f11b3f
8 changed files with 19 additions and 14 deletions
@@ -1,13 +1,13 @@
 build:
-  cuda: "11.4"
+  cuda: "11.5.1"
  gpu: true
  python_version: "3.10"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
-    - "min-dalle==0.3.5"
+    - "min-dalle==0.3.7"
  run:
-    - pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
+    - pip install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html

-predict: "replicate_predictor.py:ReplicatePredictor"
+predict: "replicate_predictor.py:ReplicatePredictor"