anaconda 安装 使用 pytorch onnx onnxruntime

 

一:安装

如果不是 x86_64,需要去镜像看对应的版本

 安装 Anaconda

输入命令

bash Anaconda3-2021.11-Linux-x86_64.sh

然后输入 yes 表示同意

确认安装的路径,一般直接回车安装在默认的 /home/你的名字/anaconda3

很快就安装完毕。输入 yes 来确认使用 conda init 来启动

3. 启动环境变量

如果现在输入 conda,会显示找不到命令

需要启动已经修改环境变量,输入以下命令(以后都不用再 source 了,因为启动 Ubuntu 会自动 source)

PyTorch 环境中安装和配置 ONNX,主要涉及以下几个步骤: ### 1. 安装 ONNX ONNX 可以通过 `pip` 或 `conda` 进行安装。建议使用 `pip` 安装最新版本的 ONNX: ```bash pip install onnx ``` 如果使用 Anaconda 环境,则可以使用以下命令: ```bash conda install -c conda-forge onnx ``` 安装完成后,可以通过以下代码验证 ONNX 是否安装成功: ```python import onnx print('ONNX 版本', onnx.__version__) ``` ### 2. 安装 ONNX Runtime ONNX Runtime 是用于运行 ONNX 模型的推理引擎。同样可以通过 `pip` 或 `conda` 安装: ```bash pip install onnxruntime ``` 如果需要 GPU 支持,则可以安装 `onnxruntime-gpu`: ```bash pip install onnxruntime-gpu ``` 验证 ONNX Runtime 是否安装成功: ```python import onnxruntime as ort print('ONNX Runtime 版本', ort.__version__) ``` ### 3. 配置 PyTorchONNX 的转换 PyTorch 提供了内置的 ONNX 导出功能,可以将训练好的模型转换为 ONNX 格式。以下是一个示例代码,展示如何将模型转换为 ONNX 格式: ```python import torch # 假设 model 是已经定义好的 PyTorch 模型 model = ... # 加载模型 model.eval() # 设置为评估模式 # 创建一个虚拟输入张量(batch_size=1, channels=3, height=256, width=256) x = torch.randn(1, 3, 256, 256) # 导出模型为 ONNX 格式 torch.onnx.export( model, x, "model.onnx", opset_version=11, input_names=['input'], output_names=['output'] ) ``` ### 4. 验证导出的 ONNX 模型 可以使用 ONNX Runtime 运行导出的模型,并验证其输出是否与原始 PyTorch 模型一致: ```python import numpy as np import onnxruntime as ort # 使用 ONNX Runtime 运行模型 ort_session = ort.InferenceSession("model.onnx") outputs = ort_session.run( None, {'input': x.numpy()} ) # 对比 ONNX Runtime 和 PyTorch 的输出 torch_output = model(x).detach().numpy() np.testing.assert_allclose(torch_output, outputs[0], rtol=1e-03, atol=1e-05) print("Exported model has been tested with ONNXRuntime, and the result looks good!") ``` ### 5. 环境配置建议 根据经验,以下是一个稳定的环境配置,适用于 PyTorch、CUDA 和 ONNX 的组合: - PyTorch 1.7.1 - CUDA 11.0.03 - cuDNN 8.0.2.39 - ONNX 1.9 - ONNX Runtime (或 ONNX Runtime-GPU) 1.8.0 如果使用更高版本的 PyTorchONNX,需要确保兼容性,尤其是 ONNX Runtime 的版本应与 ONNX 模型的 opset 版本保持一致。 ---
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值