moriire commited on
Commit
da0aaeb
·
verified ·
1 Parent(s): 1f7a1e2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -28,9 +28,9 @@ 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
 
 
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