AMCT（PyTorch）

PyTorch版本的CPU或GPU

2.1.0、1.10.0、1.8.0、1.5.0、1.4.0

安装PyTorch时，请确保pip版本低于23.0.1，否则可能出现如下“ModuleNotFoundError:No module named 'torch' ”错误信息；如果用户pip版本高于23.0.1，且不想降低版本，则请优先安装wheel包（参考安装命令：pip3 install wheel --user），然后再执行后续操作。

• 2.1.0版本PyTorch，CPU或GPU安装命令：
  • CPU版本安装命令：

    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版本安装命令：

    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

• 1.10.0版本PyTorch，CPU或GPU安装命令（推荐使用1.10.0版本）：
  • CPU版本安装命令：

    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版本安装命令：

    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

• 1.8.0版本PyTorch，CPU或GPU安装命令：
  • CPU版本安装命令：

    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版本安装命令：

    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

• 1.5.0版本PyTorch，CPU或GPU安装命令：
  • CPU版本安装命令：

    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版本安装命令：

    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

• 1.4.0版本PyTorch，CPU或GPU安装命令（不再推荐使用，使用时会有告警提示信息）：

  • CPU版本安装命令：

    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版本安装命令：

    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.14.0、1.10.0、1.9.0、1.8.0

• ONNX 1.14.0版本

  pip3 install onnx==1.14.0 --user

• ONNX 1.10.0版本

  pip3 install onnx==1.10.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.16.0、1.9.0、1.8.0、1.6.0

• ONNX Runtime 1.16.0版本

  pip3 install onnxruntime==1.16.0 --user

• ONNX Runtime 1.9.0版本

  pip3 install onnxruntime==1.9.0 --user

• ONNX Runtime1.8.0版本

  pip3 install onnxruntime==1.8.0 --user

• ONNX Runtime1.6.0版本

  pip3 install onnxruntime==1.6.0 --user

python

以3.9.2版本为例

Ubuntu操作系统请参见IFMR数据量化时，存在"inf或NaN值"或"xxx calculate scale failed"，量化过程报错。

EulerOS操作系统请参见量化过程中提示某层bias量化超出int32范围。

numpy

• PyTorch 2.1.0配套：>=1.21.6
• PyTorch 1.10.0、1.8.0、1.5.0、1.4.0配套：1.20.0~1.23.5

• PyTorch 2.1.0版本

  pip3 install numpy==1.21.6 --user

• PyTorch 1.10.0、1.8.0、1.5.0、1.4.0版本

  pip3 install numpy==1.20.0 --user

protobuf

• PyTorch 2.1.0配套：>=3.20.2
• PyTorch 1.10.0、1.8.0、1.5.0、1.4.0配套：3.13.0+

• PyTorch 2.1.0版本

  pip3 install protobuf==3.20.2 --user

• PyTorch 1.10.0、1.8.0、1.5.0、1.4.0版本

  pip3 install protobuf==3.13.0 --user