From 65d3cdcb42a798f9e5088319a3acd7f990e241c3 Mon Sep 17 00:00:00 2001 From: Hao Zhang Date: Sat, 2 Jul 2022 02:05:05 -0700 Subject: [PATCH] [ALPA-60] update with new pypi index [skip ci] (#578) --- docs/install.rst | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/install.rst b/docs/install.rst index d9e4daf2b..99ff4619d 100644 --- a/docs/install.rst +++ b/docs/install.rst @@ -64,21 +64,21 @@ If you need to use other CUDA, cuDNN, or Python versions, please follow the next pip3 install alpa -2. Then install the Alpa-modified Jaxlib from our `self-hosted PyPI server `_, - and make sure that the jaxlib version corresponds to the version of the existing CUDA and cuDNN installation you want to use. +2. Then install the Alpa-modified Jaxlib and make sure that the jaxlib version corresponds to the version of + the existing CUDA and cuDNN installation you want to use. You can specify a particular CUDA and cuDNN version for jaxlib explicitly via: .. code:: bash - pip3 install --trusted-host 169.229.48.123 -i http://169.229.48.123:8080/simple/ jaxlib==0.3.5+cuda{cuda_version}.cudnn{cudnn_version} + pip3 install jaxlib==0.3.5+cuda{cuda_version}.cudnn{cudnn_version} -f https://alpa-projects.github.io/wheels.html For example, to install the wheel compatible with CUDA >= 11.1 and cuDNN >= 8.0.5, use the following command: .. code:: bash - pip3 install --trusted-host 169.229.48.123 -i http://169.229.48.123:8080/simple/ jaxlib==0.3.5+cuda111.cudnn805 + pip3 install jaxlib==0.3.5+cuda111.cudnn805 -f https://alpa-projects.github.io/wheels.html - You can see all available wheel versions we provided at our `PyPI index `_. + You can see all available wheel versions we provided at our `PyPI index `_. .. note::