alessandro trinca tornidor
commited on
Commit
•
6d98d37
1
Parent(s):
d52c87f
[ci] try substituting nvcr.io/nvidia/pytorch:24.03-py3 with pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -10,7 +10,7 @@ ARG POETRY_CACHE_DIR=/tmp/poetry_cache
|
|
| 10 |
ARG DEPENDENCY_GROUP=fastapi
|
| 11 |
|
| 12 |
|
| 13 |
-
FROM
|
| 14 |
|
| 15 |
LABEL authors="alessandro@trinca.tornidor.com"
|
| 16 |
|
|
@@ -70,7 +70,7 @@ RUN git clone -n --depth=1 --filter=tree:0 https://huggingface.co/spaces/aletrn/
|
|
| 70 |
WORKDIR ${LAMBDA_TASK_ROOT}
|
| 71 |
RUN ls -l ${LAMBDA_TASK_ROOT}//*
|
| 72 |
|
| 73 |
-
FROM
|
| 74 |
|
| 75 |
ARG ARCH
|
| 76 |
ARG LAMBDA_TASK_ROOT
|
|
|
|
| 10 |
ARG DEPENDENCY_GROUP=fastapi
|
| 11 |
|
| 12 |
|
| 13 |
+
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime as builder_global
|
| 14 |
|
| 15 |
LABEL authors="alessandro@trinca.tornidor.com"
|
| 16 |
|
|
|
|
| 70 |
WORKDIR ${LAMBDA_TASK_ROOT}
|
| 71 |
RUN ls -l ${LAMBDA_TASK_ROOT}//*
|
| 72 |
|
| 73 |
+
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime as runtime
|
| 74 |
|
| 75 |
ARG ARCH
|
| 76 |
ARG LAMBDA_TASK_ROOT
|