Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -6,6 +6,7 @@ WORKDIR /home/user
|
|
| 6 |
|
| 7 |
COPY ./requirements.txt /home/user/requirements.txt
|
| 8 |
|
|
|
|
| 9 |
RUN pip install --no-cache-dir --upgrade -r /home/user/requirements.txt
|
| 10 |
|
| 11 |
COPY --chown=user . .
|
|
|
|
| 6 |
|
| 7 |
COPY ./requirements.txt /home/user/requirements.txt
|
| 8 |
|
| 9 |
+
RUN pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
|
| 10 |
RUN pip install --no-cache-dir --upgrade -r /home/user/requirements.txt
|
| 11 |
|
| 12 |
COPY --chown=user . .
|