Remove torch GPU dependencies
This commit is contained in:
parent
3525899277
commit
6a498f0d79
1 changed files with 2 additions and 1 deletions
|
@ -6,7 +6,8 @@ RUN apt-get update && \
|
||||||
apt-get install -y build-essential python3 python3-pip
|
apt-get install -y build-essential python3 python3-pip
|
||||||
|
|
||||||
RUN pip install --upgrade pip setuptools wheel \
|
RUN pip install --upgrade pip setuptools wheel \
|
||||||
&& pip install numpy requests sentencepiece torch tqdm
|
&& pip install numpy requests sentencepiece tqdm \
|
||||||
|
&& pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue