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
|