centos中安装onnx_graphsurgeon
时间: 2023-07-24 10:04:13 浏览: 146
Lancger#opslinux#Centos6.5安装node_exporter1
可以按照以下步骤在CentOS中安装onnx_graphsurgeon:
1. 确保您已安装Python和pip。如果没有,请使用以下命令安装:
```
sudo yum install python3 python3-pip
```
2. 安装依赖项:
```
sudo yum install -y protobuf-compiler
```
3. 使用pip安装onnx_graphsurgeon:
```
pip3 install onnx-graphsurgeon
```
4. 验证安装是否成功:
```
graphsurgeon -h
```
如果成功安装,则应该看到onnx_graphsurgeon的帮助信息。
注意:onnx_graphsurgeon需要CUDA和TensorRT支持才能在GPU上运行。如果您需要在GPU上运行,请确保已正确安装CUDA和TensorRT。
阅读全文