Documentation

AMCT (PyTorch)

Install software as the AMCT installation user. If the said user is a non-root user, check that the user has the sudo permission. Then run the su - username command to switch to the non-root user and run the commands listed in the following table. (If the installation user is root, delete --user from the dependency installation command.)

Table 1 Dependencies

Dependency

Version

Installation Command

CPU or GPU version of PyTorch

2.7.1, 2.1.0, 1.10.0, 1.8.0, 1.5.0, 1.4.0

When installing PyTorch, ensure that the pip version is earlier than 23.0.1. Otherwise, you may get the error "ModuleNotFoundError:No module named 'torch'." If the pip version is later than 23.0.1 and you do not want to downgrade the version, install the wheel package first by using the pip3 install wheel --user command and then perform subsequent operations.

  • PyTorch 2.7.1 (EulerOS (AArch64) does not support quantization on the GPU):
    python3 -m pip --trusted-host=download.pytorch.org install torch==2.7.1 torchvision==0.22.1  -f https://download.pytorch.org/whl/torch_stable.html --user
  • CPU or GPU version of PyTorch 2.1.0 (EulerOS (AArch64) does not support quantization on the GPU):
    • CPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==2.1.0+cpu torchvision==0.16.0+cpu  -f https://download.pytorch.org/whl/torch_stable.html --user
    • GPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==2.1.0 torchvision==0.16.0  -f https://download.pytorch.org/whl/torch_stable.html --user
  • (Recommended) CPU or GPU version of PyTorch 1.10.0 (EulerOS (AArch64) does not support quantization on the GPU):
    • CPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.10.0+cpu torchvision==0.11.0+cpu  -f https://download.pytorch.org/whl/torch_stable.html --user
    • GPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.10.0 torchvision==0.11.0  -f https://download.pytorch.org/whl/torch_stable.html --user
  • CPU or GPU version of PyTorch 1.8.0 (EulerOS (AArch64) does not support quantization on the GPU):
    • CPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.8.0+cpu torchvision==0.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --user
    • GPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html --user
  • CPU or GPU version of PyTorch 1.5.0 (EulerOS (AArch64) does not support quantization on the GPU):
    • CPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --user
    • GPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html --user

  • CPU or GPU version of PyTorch 1.4.0 (not recommended due to disturbing warnings):

    • CPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html --user
    • GPU version:
      python3 -m pip --trusted-host=download.pytorch.org install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html --user

ONNX

1.18.0, 1.16.1, 1.14.0, 1.9.0, 1.8.0
  • ONNX 1.18.0
    pip3 install onnx==1.18.0 --user
  • ONNX 1.16.1
    pip3 install onnx==1.16.1 --user
  • ONNX 1.14.0
    pip3 install onnx==1.14.0 --user
  • ONNX 1.9.0
    pip3 install onnx==1.9.0 --user
  • ONNX 1.8.0
    pip3 install onnx==1.8.0 --user

ONNX Runtime

1.20.0, 1.16.0, 1.8.0, 1.6.0
  • ONNX Runtime 1.20.0
    pip3 install onnxruntime==1.20.0 --user
  • ONNX Runtime 1.16.0
    pip3 install onnxruntime==1.16.0 --user
  • ONNX Runtime 1.8.0
    pip3 install onnxruntime==1.8.0 --user
  • ONNX Runtime 1.6.0
    pip3 install onnxruntime==1.6.0 --user

Python

Version 3.9.2 is used as an example.

For Ubuntu OS, see Python 3.9.2 Installation on Ubuntu.

For EulerOS, see Python3.9.2 Installation on EulerOS.

NumPy
  • PyTorch 2.7.1 and Python 3.12/3.13 work with NumPy 2.0.0 and later versions.
  • PyTorch 2.7.1 and Python 3.11.x work with NumPy 1.24.2–1.26.4.
  • PyTorch 2.1.0 works with NumPy 1.21.6 and later versions.
  • PyTorch 1.10.0, 1.8.0, 1.5.0, and 1.4.0 work with NumPy 1.20.0–1.23.5.
  • PyTorch 2.7.1
    pip3 install numpy==1.24.2 --user
  • PyTorch 2.1.0
    pip3 install numpy==1.21.6 --user
  • PyTorch 1.10.0, 1.8.0, 1.5.0, and 1.4.0
    pip3 install numpy==1.20.0 --user

Protobuf
  • PyTorch 2.7.1 and Python 3.12/3.13 work with Protobuf 3.20.3.
  • PyTorch 2.7.1 and Python 3.11.x work with Protobuf 3.20.2–3.20.3.
  • PyTorch 2.1.0, 1.10.0, 1.8.0, 1.5.0, and 1.4.0 work with Protobuf 3.20.2 and later versions.
 
pip3 install protobuf==3.20.3 --user

Or

pip3 install protobuf==3.20.2 --user
Parent topic: Installing Dependencies
This site uses cookies to improve your browsing experience. By continuing to browse this site, you agree to our use of cookies.View Details