ubuntu安装onnxruntime
时间: 2023-05-02 13:02:08 浏览: 1672
安装ONNX Runtime可以按照以下步骤进行:
1. 下载ONNX Runtime的安装包,可以到官网下载:https://onnxruntime.ai/
2. 打开终端,进入到下载的安装包文件夹中,执行以下命令:
```
sudo apt-get update && sudo apt-get install libprotobuf-dev protobuf-compiler libgomp1
sudo dpkg -i onnxruntime-x.x.x-cpu.deb
```
(x.x.x为所下载的安装包版本号)
3. 安装完成后,可以在终端中输入以下命令验证ONNX Runtime是否安装成功:
```
python -c "import onnxruntime"
```
如果没有报错,则说明ONNX Runtime已经成功安装,可以开始使用了。
相关问题
ubuntu18.04安装onnxruntime-gpu
可以按照以下步骤安装onnxruntime-gpu:
1. 确保你的GPU支持CUDA和cuDNN,并且已经安装了CUDA和cuDNN。
2. 安装依赖项:
```
sudo apt-get update
sudo apt-get install -y git cmake build-essential libprotobuf-dev protobuf-compiler libprotobuf-dev libprotoc-dev libopencv-dev
```
3. 克隆onnxruntime-gpu仓库:
```
git clone --recursive https://github.com/microsoft/onnxruntime.git
```
4. 进入onnxruntime目录:
```
cd onnxruntime
```
5. 创建build目录并进入:
```
mkdir build
cd build
```
6. 配置编译选项:
```
cmake .. -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/usr/local -Donnxruntime_USE_CUDA=ON -Donnxruntime_CUDA_HOME=/usr/local/cuda -Donnxruntime_CUDA_INCLUDE_DIRS=/usr/local/cuda/include -Donnxruntime_CUDA_LIB_DIRS=/usr/local/cuda/lib64 -Donnxruntime_USE_TENSORRT=OFF
```
其中,-Donnxruntime_USE_CUDA=ON表示使用CUDA,-Donnxruntime_CUDA_HOME、-Donnxruntime_CUDA_INCLUDE_DIRS、-Donnxruntime_CUDA_LIB_DIRS分别指定CUDA的安装路径和头文件路径、库文件路径。
7. 编译并安装:
```
make -j$(nproc)
sudo make install
```
这里使用了make -j$(nproc)来并行编译,可以加快编译速度。
8. 安装完毕后,可以测试一下onnxruntime-gpu是否正常工作:
```
cd ../tools/ci_build/github/linux/docker/scripts
./run_onnx_tests.sh -e gpu
```
如果测试通过,则说明onnxruntime-gpu已经安装成功了。
注意:onnxruntime-gpu的安装可能会因为各种原因失败,需要根据具体错误信息进行调试。
ubuntu系统安装onnxruntime Defaulting to user installation because normal site-packages is not writeable
这个问题通常是由于权限问题导致的。您可以尝试以下解决方法:
1. 确保您以管理员身份运行安装命令。
2. 如果您使用的是虚拟环境,请确保已经激活该环境,并且您有足够的权限在环境中进行安装。
3. 尝试在命令前加上sudo,这将以管理员权限运行命令。
4. 如果上述方法都不起作用,您可以尝试将python安装的位置更改为您有权限的目录。您可以使用以下命令检查您的python安装位置:
```
which python
```
然后,您可以使用以下命令更改安装位置:
```
pip install --user onnxruntime
```
这将在用户目录下安装onnxruntime。
阅读全文