Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
438c2df
1
Parent(s):
d6996ca
Update Dockerfile
Browse files- Dockerfile +1 -2
Dockerfile
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
-
FROM
|
2 |
WORKDIR /app
|
3 |
|
4 |
COPY ./requirements.txt /app/requirements.txt
|
5 |
COPY ./packages.txt /app/packages.txt
|
6 |
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
|
7 |
-
RUN pip install Jinja2
|
8 |
RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
|
9 |
RUN pip3 install --no-cache-dir -r /app/requirements.txt
|
10 |
|
|
|
1 |
+
FROM anibali/pytorch:1.13.0-cuda11.8
|
2 |
WORKDIR /app
|
3 |
|
4 |
COPY ./requirements.txt /app/requirements.txt
|
5 |
COPY ./packages.txt /app/packages.txt
|
6 |
RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
|
|
|
7 |
RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
|
8 |
RUN pip3 install --no-cache-dir -r /app/requirements.txt
|
9 |
|