From 8b76cb809501028fc2f6d2565f81a2097ebd4e88 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Fri, 29 Jan 2021 13:35:16 +0100 Subject: [PATCH] Rephrase transformers PyTorch instructions --- website/docs/usage/embeddings-transformers.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/website/docs/usage/embeddings-transformers.md b/website/docs/usage/embeddings-transformers.md index ce1b63dfd..e71336e84 100644 --- a/website/docs/usage/embeddings-transformers.md +++ b/website/docs/usage/embeddings-transformers.md @@ -204,8 +204,7 @@ drop-in replacements that let you achieve **higher accuracy** in exchange for > downloaded: 3GB CUDA runtime, 800MB PyTorch, 400MB CuPy, 500MB weights, 200MB > spaCy and dependencies. -Once you have CUDA installed, we recommend installing PyTorch separately -following the +Once you have CUDA installed, we recommend installing PyTorch following the [PyTorch installation guidelines](https://pytorch.org/get-started/locally/) for your package manager and CUDA version. If you skip this step, pip will install PyTorch as a dependency below, but it may not find the best version for your @@ -213,6 +212,7 @@ setup. ```bash ### Example: Install PyTorch 1.7.1 for CUDA 10.1 with pip +# See: https://pytorch.org/get-started/locally/ $ pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html ```