OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[Doc] update default magma cuda version in readme (#122125)

Browse Source 
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
This commit is contained in:
youkaichao 2024-05-28 03:37:23 +00:00 committed by PyTorch MergeBot
parent 7c61e7be5c
commit d957c2d5de
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -209,7 +209,7 @@ pip install -r requirements.txt
```bash ```bash
conda install intel::mkl-static intel::mkl-include conda install intel::mkl-static intel::mkl-include
# CUDA only: Add LAPACK support for the GPU if needed # 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 # (optional) If using torch.compile with inductor/triton, install the matching version of triton
# Run from the pytorch directory after cloning # Run from the pytorch directory after cloning