mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
c0be0105c7
Add NVPL to CUDA ARM docker build original https://github.com/pytorch/builder/pull/1928 moving to pytorch/pytorch repo now Need to go with builder repo change https://github.com/pytorch/builder/pull/1950 Pull Request resolved: https://github.com/pytorch/pytorch/pull/132811 Approved by: https://github.com/atalman
20 lines
773 B
Bash
20 lines
773 B
Bash
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
function install_nvpl {
|
|
|
|
mkdir -p /opt/nvpl/lib /opt/nvpl/include
|
|
|
|
wget https://developer.download.nvidia.com/compute/nvpl/redist/nvpl_blas/linux-sbsa/nvpl_blas-linux-sbsa-0.3.0-archive.tar.xz
|
|
tar xf nvpl_blas-linux-sbsa-0.3.0-archive.tar.xz
|
|
cp -r nvpl_blas-linux-sbsa-0.3.0-archive/lib/* /opt/nvpl/lib/
|
|
cp -r nvpl_blas-linux-sbsa-0.3.0-archive/include/* /opt/nvpl/include/
|
|
|
|
wget https://developer.download.nvidia.com/compute/nvpl/redist/nvpl_lapack/linux-sbsa/nvpl_lapack-linux-sbsa-0.2.3.1-archive.tar.xz
|
|
tar xf nvpl_lapack-linux-sbsa-0.2.3.1-archive.tar.xz
|
|
cp -r nvpl_lapack-linux-sbsa-0.2.3.1-archive/lib/* /opt/nvpl/lib/
|
|
cp -r nvpl_lapack-linux-sbsa-0.2.3.1-archive/include/* /opt/nvpl/include/
|
|
}
|
|
|
|
install_nvpl |