window 安装onnx和onnxruntime-gpu（python版）

1. conda_py37.yaml文件

2. 命令行安装

3. 效果

4. 代码测试环境

配置环境cuda10.2，onnx1.8.0，onnxruntime-gpu=1.6.0

其他版本对应关系：

CUDA - onnxruntimeInstructions to execute ONNX Runtime applications with CUDAhttps://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html

1. conda_py37.yaml文件

name: test_onnx
channels:- Esri- conda-forge- defaults
dependencies:- _pytorch_select=1.2.0=gpu- blas=1.0=mkl- ca-certificates=2021.5.25=haa95532_1- certifi=2021.5.30=py37haa95532_0- cffi=1.14.5=py37hd8e9650_0- cudatoolkit=10.2.89=h74a9793_1- intel-openmp=2019.4=245- libblas=3.8.0=14_mkl- libcblas=3.8.0=14_mkl- liblapack=3.8.0=14_mkl- libmklml=2019.0.5=haa95532_0- libprotobuf=3.14.0=h7755175_0- mkl=2019.4=245- mkl-service=2.3.0=py37hfa6e2cd_0- ninja=1.10.2=h5362a0b_0- numpy=1.20.3=py37hcbcd69c_1- onnx=1.8.0=py37hb2e8ee1_1- openssl=1.1.1k=h2bbff1b_0- pip=21.1.2=pyhd8ed1ab_0- protobuf=3.14.0=py37hf2a7229_1- pycparser=2.20=pyh9f0ad1d_2- python=3.7.10=0- python_abi=3.7=1_cp37m- setuptools=49.6.0=py37h03978a9_3- six=1.15.0=py_0- sqlite=3.35.5=h8ffe710_0- tbb=2021.2.0=h2d74725_0- typing-extensions=3.10.0.0=hd8ed1ab_0- typing_extensions=3.10.0.0=pyha770c72_0- vc=14.1=h869be7e_1- vs2015_runtime=14.16.27012=hf0eaf9b_0- wheel=0.36.2=pyhd3deb0d_0- wincertstore=0.2=py37h03978a9_1006- zlib=1.2.11=h62dcd97_1010- pip:- altgraph==0.17- flatbuffers==2.0- future==0.18.2- importlib-metadata==4.5.0- onnxruntime-gpu==1.6.0- opencv-contrib-python==4.5.2.52- opencv-python==4.5.2.52- pefile==2021.5.24- pyinstaller==4.3- pyinstaller-hooks-contrib==2021.1- pylibdmtx==0.1.9- pywin32-ctypes==0.2.0- zipp==3.4.1
prefix: D:\ProgramData\Anaconda3\envs\test_onnx

2. 命令行安装

# 安装
conda env create -f conda_py37.yaml
# 激活环境
conda activate test_onnx
# 确认环境
conda list

3. 效果

4. 代码测试环境

下载onnx权重进行测试

alexnet.onnx权重-互联网文档类资源-CSDN下载生成该权重代码：importtorchimporttorchvision#useTr更多下载资源、学习资料请访问CSDN下载频道.https://download.csdn.net/download/jizhidexiaoming/80233035

import timeimport onnx
import numpy as np
import onnxruntime as ort# load onnx model
model = onnx.load_model('alexnet.onnx')
onnx.checker.check_model(model=model)
onnx.helper.printable_graph(graph=model.graph)# onnx_runtime val
# sess_options = ort.SessionOptions()
# sess_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_DISABLE_ALL
# sess_options.optimized_model_filepath = "optimized_mnist_original.onnx"
# session = ort.InferenceSession("mnist_original.onnx", sess_options)
ort_session = ort.InferenceSession("alexnet.onnx")
while 1:start = time.time()outputs = ort_session.run(None, {'inputs': np.random.randn(10, 3, 224, 224).astype(np.float32)})  # inputs is same to origprint(time.time() - start)

window 安装onnx和onnxruntime-gpu（python版）相关推荐

win10安装onnx、tensorrt（python用，超简单安装版)
一. 安装环境版本显卡是3090. 1. python 3.8 2. pytorch 1.9.1(torch-1.9.1+cu111-cp38-cp38-win_amd64),离线安装参考:(10条 ...
linux安装pangolin python版
linux安装pangolin python版安装 pangolin 依赖项安装报错测试效果 pangolin pangolin是一个轻量的3D库,可用于3维重建.SLAM的中点云数据的可视化 ...
HP Z840 安装Ubuntu18.04，gpu版的tensorflow，keras全程纪要，硬件Qurado P4000两块
HP Z840 安装Ubuntu18.04,gpu版的tensorflow,keras全程纪要,硬件Qurado P4000两块 Step1:安装SSH协议并开启(方便远程安装) sudo apt-g ...
保姆级教程——Ubuntu16.04 Server下深度学习环境搭建：安装CUDA8.0，cuDNN6.0，Bazel0.5.4，源码编译安装TensorFlow1.4.0(GPU版)...
写在前面本文叙述了在Ubuntu16.04 Server下安装CUDA8.0,cuDNN6.0以及源码编译安装TensorFlow1.4.0(GPU版)的亲身经历,包括遇到的问题及解决办法,也有一些 ...
*** Python版一键安装脚本
本脚本适用环境: 系统支持:CentOS 6,7,Debian,Ubuntu 内存要求:≥128M 日期:2018 年 02 月 07 日关于本脚本: 一键安装 Python 版 *** 的最新版. ...
1-3.Win10系统利用Pycharm社区版安装Django搭建一个简单Python Web项目的步骤之三
在1-1.Win10系统利用Pycharm社区版安装Django搭建一个简单Python Web项目的步骤之一基础上进行如下操作: 所有路由不能全部都在myDjango下的urls.py路由文件中, ...
在mac上安装python版的hanlp/JPype1
安装python版的hanlp时老是出错,直接 pip install pyhanlp,会出现这个错误: In file included from native/common/jp_method.c ...
Python ln_虚拟机中单机Spark （Python版）安装
Spark 安装成功界面检验Spark(Python版)是否安装成功: # pyspark 楼主已经安装完虚拟机,以下操作为单机版Python版Spark安装步骤: *安装虚拟机 *安装java 检 ...
本地python版GEE安装及配置步骤（jupyter lab)
之前一直使用的是javaScript版的GEE,方便是方便,但我想完成的功能大概还是得借助python吧.于是花了好几天从零开始安装配置.虽然路上遇到好多问题,但好在最终结果还不错. 发现好像pyth ...

window 安装onnx和onnxruntime-gpu（python版）

1. conda_py37.yaml文件

2. 命令行安装

3. 效果

4. 代码测试环境

window 安装onnx和onnxruntime-gpu（python版）相关推荐

最新文章

热门文章