From de83f3f905ef8b4ac3a6c6e068f98fdd66eed195 Mon Sep 17 00:00:00 2001 From: Iulia Feroli Date: Thu, 21 Mar 2024 11:21:32 +0100 Subject: [PATCH] Improve PyTorch installation instructions (#677) --- README.md | 4 ++-- docs/guide/machine-learning.asciidoc | 7 +++---- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index 180ac3c..ea0ad85 100644 --- a/README.md +++ b/README.md @@ -58,8 +58,8 @@ $ conda install -c conda-forge eland If you are using the NLP with PyTorch feature make sure your Eland minor version matches the minor version of your Elasticsearch cluster. For all other features it is sufficient for the major versions to match. -- You need to use PyTorch `1.13.1` or earlier to import an NLP model. - Run `pip install torch==1.13.1` to install the aproppriate version of PyTorch. +- You need to install the appropriate version of PyTorch to import an NLP model. Run `python -m pip + install 'eland[pytorch]'` to install that version. ### Prerequisites diff --git a/docs/guide/machine-learning.asciidoc b/docs/guide/machine-learning.asciidoc index 5c944d1..eee3b86 100644 --- a/docs/guide/machine-learning.asciidoc +++ b/docs/guide/machine-learning.asciidoc @@ -38,9 +38,8 @@ model in {es}. [[ml-nlp-pytorch]] === Natural language processing (NLP) with PyTorch - -IMPORTANT: You need to use PyTorch `1.13` or earlier to import an NLP model. -Run `pip install torch==1.13` to install the aproppriate version of PyTorch. +IMPORTANT: You need to install the appropriate version of PyTorch to import an +NLP model. Run `python -m pip install 'eland[pytorch]'` to install that version. For NLP tasks, Eland enables you to import PyTorch models into {es}. Use the `eland_import_hub_model` script to download and install supported @@ -191,4 +190,4 @@ eland_import_hub_model --es-api-key --url https://: .. -------------------------------------------------- eland_import_hub_model --hub-access-token ... -------------------------------------------------- --- \ No newline at end of file +--