31 lines
846 B
Plaintext
31 lines
846 B
Plaintext
[
|
|
{ type: install
|
|
message: <<EOM
|
|
You installed PyTorch: an AI toolkit.
|
|
|
|
## Need env var on amd64 Zen5/Zen6 CPU architectures
|
|
|
|
Due to the bug in the OpenBLAS CPU arch detection PyTorch will
|
|
crash on Zen5/Zen6 CPU architectures.
|
|
|
|
Reference: https://github.com/OpenMathLib/OpenBLAS/issues/6029
|
|
|
|
Use the following environment variable to work around this problem:
|
|
OPENBLAS_CORETYPE=SKYLAKEX (some other values might work too)
|
|
|
|
The same problem exists in all software that uses OpenBLAS, not just PyTorch.
|
|
|
|
## Need env var on aarch64 due to high TLS requirements
|
|
|
|
When running PyTorch on aarch64 you will need to have this environment
|
|
variable set:
|
|
LD_STATIC_TLS_EXTRA=4096
|
|
|
|
It pre-allocates more TLS space when the Python process starts so that
|
|
the PyTorch shared libraries will have sufficient TLS space when they are
|
|
loaded dynamically.
|
|
|
|
EOM
|
|
}
|
|
]
|