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 of the PyTorch version

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.

  • Installation command for the CPU version of PyTorch 2.7.1
    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
  • Installation command for the CPU version of PyTorch 2.1.0
    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
  • Installation command for the CPU version of PyTorch 1.10.0 (recommended)
    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
  • Installation command for the CPU version of PyTorch 1.8.0
    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
  • Installation command for the CPU version of PyTorch 1.5.0
    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
  • Installation command for the CPU version of PyTorch 1.4.0 (not recommended, as a warning message will be displayed when it is used)
    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

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