文档HOT
下载
中文
注册

AMCT（PyTorch）

请使用AMCT的安装用户安装依赖的软件，如果安装用户为非root，请确保该用户拥有sudo权限，请使用su - username命令切换到非root用户执行如下命令。

表1 依赖列表

依赖名称

版本号

安装命令

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操作系统请参见安装Python3.9.2（Ubuntu）。

EulerOS操作系统请参见安装Python3.9.2（EulerOS ）。

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
父主题： 安装依赖