ubuntu18.04 cuda11.3 安装哪个onnxrunrtime

您好，针对您的操作系统和CUDA版本，可以安装ONNX Runtime 1.8.2版本。您可以通过以下步骤安装：

1. 安装依赖项：

sudo apt-get update
sudo apt-get install -y curl libgomp1

2. 下载ONNX Runtime安装包：

curl https://aka.ms/onnxruntime-server-linux-x64-1.8.2 -o onnxruntime-server-linux-x64-1.8.2.tgz

3. 解压安装包：

tar -xzf onnxruntime-server-linux-x64-1.8.2.tgz

4. 将ONNX Runtime添加到系统路径中：

echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:'`pwd`'/onnxruntime-server-linux-x64-1.8.2/lib' >> ~/.bashrc
source ~/.bashrc

安装完成后，您可以在Python中使用ONNX Runtime。例如，您可以使用以下代码测试ONNX Runtime是否正常工作：

import onnxruntime

print("ONNX Runtime version: ", onnxruntime.__version__)

# Load an ONNX model
sess = onnxruntime.InferenceSession("model.onnx")

# Get the input name
input_name = sess.get_inputs()[0].name

# Create a random input tensor
import numpy as np
input_data = np.random.rand(1, 3, 224, 224).astype(np.float32)

# Run the model
output = sess.run(None, {input_name: input_data})

print("Model output shape: ", output[0].shape)

希望这能帮助到您！