Remove pinning & use CPU whls directly
Co-authored-by: Pavol Rusnak <pavol@rusnak.io>
This commit is contained in:
parent
6a498f0d79
commit
9fe1524d02
1 changed files with 1 additions and 1 deletions
|
@ -7,7 +7,7 @@ RUN apt-get update && \
|
|||
|
||||
RUN pip install --upgrade pip setuptools wheel \
|
||||
&& pip install numpy requests sentencepiece tqdm \
|
||||
&& pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
|
||||
&& pip install torch --index-url https://download.pytorch.org/whl/cpu
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue