mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Remove mention of magma-cuda in readme.md, refactor magma_conda install (#147476)
Related to: https://github.com/pytorch/pytorch/issues/138506 we migrated magma-cuda build from anaconda to aws Last version of magma-cuda published was 12.6 https://anaconda.org/pytorch/magma-cuda126 Here is the PR that moved from anaconda to tarball: https://github.com/pytorch/pytorch/pull/140417 Pull Request resolved: https://github.com/pytorch/pytorch/pull/147476 Approved by: https://github.com/albanD
This commit is contained in:
committed by
PyTorch MergeBot
parent
c0b70f94e2
commit
43de9b75c3
@ -243,7 +243,8 @@ pip install -r requirements.txt
|
||||
```bash
|
||||
pip install mkl-static mkl-include
|
||||
# CUDA only: Add LAPACK support for the GPU if needed
|
||||
conda install -c pytorch magma-cuda121 # or the magma-cuda* that matches your CUDA version from https://anaconda.org/pytorch/repo
|
||||
# magma installation: run with active conda environment. specify CUDA version to install
|
||||
.ci/docker/common/install_magma_conda.sh 12.4
|
||||
|
||||
# (optional) If using torch.compile with inductor/triton, install the matching version of triton
|
||||
# Run from the pytorch directory after cloning
|
||||
|
Reference in New Issue
Block a user