ONNXRuntime与CUDA版本对应

版本

onnxruntime-gpu版本可以说是一个非常简单易用的框架,因为通常用pytorch训练的模型,在部署时,会首先转换成onnx,而onnxruntime和onnx又是有着同一个爸爸,无疑,在op的支持上肯定是最好的。

        通常在安装onnxruntime时,需要将其版本与pytorch版本和CUDA版本进行对应,其中ONNXRuntime与CUDA版本对应关系表如下表所示。

 选好版本后,使用以下命令进行安装onnxruntime

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

查看是否使用上GPU

python 3.8.17 (default, Jul  5 2023, 21:04:15) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import onnxruntime
>>> onnxruntime.get_device()
'GPU'
>>> onnxruntime.get_available_providers()
['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'AzureExecutionProvider', 'CPUExecutionProvider']
 

相关推荐

  1. pytorchcuda版本对应关系汇总

    2024-07-20 15:46:04       54 阅读
  2. 下降npm版本 Node版本NPM版本对应关系

    2024-07-20 15:46:04       41 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-20 15:46:04       106 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-20 15:46:04       116 阅读
  3. 在Django里面运行非项目文件

    2024-07-20 15:46:04       95 阅读
  4. Python语言-面向对象

    2024-07-20 15:46:04       103 阅读

热门阅读

  1. 素数极差

    2024-07-20 15:46:04       23 阅读
  2. 数据结构——栈

    2024-07-20 15:46:04       25 阅读
  3. 量化交易对短期收益的提升效果

    2024-07-20 15:46:04       25 阅读
  4. ArcGIS Pro SDK (九)几何 9 立方贝塞尔线段

    2024-07-20 15:46:04       23 阅读
  5. glibc: getifaddrs_internal 占用大量cpu

    2024-07-20 15:46:04       23 阅读
  6. 【关于使用swoole的知识点整理】

    2024-07-20 15:46:04       21 阅读
  7. 弹框管理类demo

    2024-07-20 15:46:04       24 阅读
  8. 单机 Redission 存在的问题以及怎么解决

    2024-07-20 15:46:04       24 阅读
  9. 力扣(LeetCode)——70. 爬楼梯

    2024-07-20 15:46:04       23 阅读