Update Dockerfile
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -24,11 +24,13 @@ WORKDIR /code
|
|
| 24 |
|
| 25 |
RUN chmod 777 .
|
| 26 |
|
| 27 |
-
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 28 |
COPY ./requirements.txt /code/requirements.txt
|
| 29 |
|
|
|
|
| 30 |
|
| 31 |
-
RUN
|
|
|
|
|
|
|
| 32 |
|
| 33 |
RUN useradd -m -u 1000 user
|
| 34 |
|
|
|
|
| 24 |
|
| 25 |
RUN chmod 777 .
|
| 26 |
|
|
|
|
| 27 |
COPY ./requirements.txt /code/requirements.txt
|
| 28 |
|
| 29 |
+
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 30 |
|
| 31 |
+
RUN pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cpu \
|
| 32 |
+
pip install intel-extension-for-pytorch==2.1.100\
|
| 33 |
+
pip install oneccl_bind_pt==2.1.0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
|
| 34 |
|
| 35 |
RUN useradd -m -u 1000 user
|
| 36 |
|