onnxruntime下载心得

onnxruntime的下载方式

一共有三种，分别为
1.python原生下载
2.虚拟机下载
3.wsl2下载
主要介绍第一种和第三种

python原生下载

首先确保已下载pip3
执行命令

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple onnxruntime

这里用到了清华园的镜像，会快一点，这里下载的是onnxruntime CPU版本的，如果要下载GPU版本，把onnxruntime换成onnxruntime-gpu即可。

安装 onnxruntime-gpu 注意事项：

1. onnxruntime-gpu包含onnxruntime的大部分功能。如果已安装onnruntime要把onnruntime卸载掉。
2. 安装时一定要注意与CUDA、cuDNN版本适配问题，具体适配列表参考：CUDA Execution Provider

安装好后验证onnxruntime是否用到GPU：

>>> import onnxruntime
>>> onnxruntime.get_device()
'GPU'  #表示GPU可用
>>> onnxruntime.get_available_providers()
['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']

wsl2下载

首先下载好wsl2的相关软件：具体下载点这里
剩下的下载就是在linux系统下的下载了。
首先下载好python：再贴上链接
然后下载好pip3

sudo apt install python3-pip

在这一步下载的时候出现了bug，如下：

原因是ubuntu没有更新,执行下面的代码进行更新

sudo apt update

然后检验pip3是否下载成功

pip3

结果如下：
这样就说明pip3下载成功
最后还是和上面一样，执行代码：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple onnxruntime

下载成功：