From d957c2d5de004083b404031bbcbee21dac52c22f Mon Sep 17 00:00:00 2001 From: youkaichao Date: Tue, 28 May 2024 03:37:23 +0000 Subject: [PATCH] [Doc] update default magma cuda version in readme (#122125) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Since we use cuda 12.1 by default now, it would be better to update the doc. Many people (including me), want to directly copy-paste commands in readme 😉 Let's make our life easier. Pull Request resolved: https://github.com/pytorch/pytorch/pull/122125 Approved by: https://github.com/malfet --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index eb291b1c97e..2a469af7b16 100644 --- a/README.md +++ b/README.md @@ -209,7 +209,7 @@ pip install -r requirements.txt ```bash conda install intel::mkl-static intel::mkl-include # CUDA only: Add LAPACK support for the GPU if needed -conda install -c pytorch magma-cuda110 # or the magma-cuda* that matches your CUDA version from https://anaconda.org/pytorch/repo +conda install -c pytorch magma-cuda121 # or the magma-cuda* that matches your CUDA version from https://anaconda.org/pytorch/repo # (optional) If using torch.compile with inductor/triton, install the matching version of triton # Run from the pytorch directory after cloning