update cuda version in replicate docker image

2022-07-05 15:07:32 -04:00
parent 3aa67b4f5d
commit cc0450337c
4 changed files with 5 additions and 6 deletions
@@ -1,13 +1,13 @@
 build:
-  cuda: "11.0"
+  cuda: "11.5.1"
  gpu: true
-  python_version: "3.8"
+  python_version: "3.10"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "min-dalle==0.2.27"
  run:
-    - pip install torch==1.10.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"