Use an image with earlier drivers

This commit is contained in:
doctorpangloss 2024-06-28 18:15:33 -07:00
parent 8f1071ea4d
commit 1980c5f236
2 changed files with 2 additions and 2 deletions

View File

@ -23,7 +23,7 @@ jobs:
- labels: [self-hosted, Linux, X64, rocm-7600-8gb]
container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
- labels: [self-hosted, Linux, X64, cuda-3060-12gb]
container: "nvcr.io/nvidia/pytorch:24.04-py3"
container: "nvcr.io/nvidia/pytorch:24.03-py3"
steps:
- run: |
apt update

View File

@ -1,4 +1,4 @@
FROM nvcr.io/nvidia/pytorch:24.04-py3
FROM nvcr.io/nvidia/pytorch:24.03-py3
RUN pip install --no-cache --no-build-isolation git+https://github.com/hiddenswitch/ComfyUI.git
EXPOSE 8188
WORKDIR /workspace