AMCT(PyTorch)
请使用AMCT的安装用户安装依赖的软件,如果安装用户为非root,请确保该用户拥有sudo权限,请使用su - username命令切换到非root用户执行如下命令(如果安装用户为root,请将安装依赖命令中的--user删除)。
依赖名称 |
版本号 |
安装命令 |
|---|---|---|
PyTorch版本的CPU |
2.7.1、2.1.0、1.10.0、1.8.0、1.5.0、1.4.0 |
|
ONNX |
1.18.0、1.16.1、1.14.0、1.9.0、1.8.0 |
|
ONNX Runtime |
1.20.0、1.16.0、1.8.0、1.6.0 |
|
Python |
以3.9.2版本为例 |
Ubuntu操作系统请参见安装Python3.9.2(Ubuntu)。 EulerOS操作系统请参见安装Python3.9.2(EulerOS )。 |
TorchNPU |
- |
|
numpy |
|
|
protobuf |
|
pip3 install protobuf==3.20.3 --user 或 pip3 install protobuf==3.20.2 --user |
pyyaml |
- |
pip3 install pyyaml --user |
transformers |
5.5.4 |
pip3 install transformers==5.5.4 --user |
loguru |
- |
pip3 install loguru --user |
accelerate |
- |
pip3 install accelerate --user |
compressed_tensors |
0.15.0.1 |
pip3 install compressed_tensors==0.15.0.1 --user |
scipy |
- |
pip3 install scipy --user |
torchao |
- |
pip3 install torchao --user |
einops |
- |
pip3 install einops --user |
sentencepiece |
- |
pip3 install sentencepiece --user |
datasets |
4.8.4 |
pip3 install datasets==4.8.4 --user |
zstandard |
0.25.0 |
pip3 install zstandard==0.25.0 --user |
父主题: 安装依赖