mirror of
https://github.com/elastic/eland.git
synced 2025-07-11 00:02:14 +08:00
Upgrade PyTorch to 2.5.1 (#785)
PyTorch was upgraded to 2.5.1 in ml-cpp on the 8.18 and 9.0 branches in elastic/ml-cpp#2800
This commit is contained in:
parent
ec45c395fd
commit
878cde6126
@ -18,7 +18,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
||||||
python3 -m pip install \
|
python3 -m pip install \
|
||||||
--no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu \
|
--no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu \
|
||||||
torch==2.3.1+cpu .[all]; \
|
torch==2.5.1+cpu .[all]; \
|
||||||
else \
|
else \
|
||||||
python3 -m pip install \
|
python3 -m pip install \
|
||||||
--no-cache-dir --disable-pip-version-check \
|
--no-cache-dir --disable-pip-version-check \
|
||||||
|
@ -13,7 +13,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
|
|||||||
if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
|
||||||
python3 -m pip install \
|
python3 -m pip install \
|
||||||
--no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu \
|
--no-cache-dir --disable-pip-version-check --extra-index-url https://download.pytorch.org/whl/cpu \
|
||||||
torch==2.3.1+cpu .[all]; \
|
torch==2.5.1+cpu .[all]; \
|
||||||
else \
|
else \
|
||||||
python3 -m pip install \
|
python3 -m pip install \
|
||||||
--no-cache-dir --disable-pip-version-check \
|
--no-cache-dir --disable-pip-version-check \
|
||||||
|
2
setup.py
2
setup.py
@ -60,7 +60,7 @@ extras = {
|
|||||||
"lightgbm": ["lightgbm>=3,<5"],
|
"lightgbm": ["lightgbm>=3,<5"],
|
||||||
"pytorch": [
|
"pytorch": [
|
||||||
"requests<3",
|
"requests<3",
|
||||||
"torch==2.3.1",
|
"torch==2.5.1",
|
||||||
"tqdm",
|
"tqdm",
|
||||||
"sentence-transformers>=2.1.0,<=2.7.0",
|
"sentence-transformers>=2.1.0,<=2.7.0",
|
||||||
# sentencepiece is a required dependency for the slow tokenizers
|
# sentencepiece is a required dependency for the slow tokenizers
|
||||||
|
Loading…
x
Reference in New Issue
Block a user