mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
magma-cuda should reference updated versions (#12000)
Summary: Source build doc section **LAPACK GPU** only lists magma-cuda80 The magma-cuda version should reflect the installed version of cuda. - Verified on ubuntu with magma-cuda92 with build and test - Verified 91 is available Pull Request resolved: https://github.com/pytorch/pytorch/pull/12000 Differential Revision: D10024158 Pulled By: soumith fbshipit-source-id: a34c85a5e87b52657f1e6f7b21d235306ab7b2aa
This commit is contained in:
parent
5d4624a1d9
commit
86e025fca2
|
|
@ -163,7 +163,7 @@ conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
|
|||
conda install -c mingfeima mkldnn
|
||||
|
||||
# Add LAPACK support for the GPU
|
||||
conda install -c pytorch magma-cuda80 # or magma-cuda90 if CUDA 9
|
||||
conda install -c pytorch magma-cuda92 # or [magma-cuda80 | magma-cuda91] depending on your cuda version
|
||||
```
|
||||
|
||||
On macOS
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user