Spaces:
Sleeping
Sleeping
cyberspyde
commited on
Commit
•
63bd0c7
1
Parent(s):
1203715
torch
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
@@ -6,6 +6,8 @@ COPY ./requirements.txt /code/requirements.txt
|
|
6 |
|
7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
8 |
|
|
|
|
|
9 |
RUN apt-get -y update
|
10 |
|
11 |
RUN useradd -m -u 1000 user
|
|
|
6 |
|
7 |
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
8 |
|
9 |
+
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
10 |
+
|
11 |
RUN apt-get -y update
|
12 |
|
13 |
RUN useradd -m -u 1000 user
|