易百纳玄武S3_模型转换_环境搭建指南
一、简介本文主要介绍易百纳玄武S3上RK182X模型转化环境搭建, 主要涉及CUDA环境搭建以及python依赖环境搭建。二、硬件要求准备一台带有NVIDIA显卡的X86架构计算机模型转换需要CUDA环境。三、系统要求Ubuntu22.04系统四、源码获取模型转换主要涉及两个仓库代码https://gitee.com/xuanwu_s3/rknn3-toolkithttps://gitee.com/xuanwu_s3/rknn3-model-zoo使用git获取git clone https://gitee.com/xuanwu_s3/rknn3-toolkit git clone https://gitee.com/xuanwu_s3/rknn3-model-zoo五、CUDA环境搭建以RTX3000为例1安装显卡驱动查看显卡设备sudo ubuntu-drivers devices安装显卡驱动sudo ubuntu-drivers autoinstall sudo reboot查看驱动是否安装好:nvidia-smi安装成功后如下图所示2安装CUDA环境下载并安装 NVIDIA 官方仓库密钥:wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb sudo dpkg -i cuda-keyring_1.1-1_all.deb更新软件源:sudo apt update安装CUDA工具包:需要安装的cuda版本是通过nvidia-smi命令输出的CUDA Version:决定这里是13.3如图所示sudo apt -y install cuda-toolkit-13-2配置cuda环境变量echo export PATH/usr/local/cuda-13.2/bin:$PATH ~/.bashrc echo export LD_LIBRARY_PATH/usr/local/cuda-13.2/lib64:$LD_LIBRARY_PATH ~/.bashrc echo export CUDA_HOME/usr/local/cuda-13.2 ~/.bashrc source ~/.bashrc检查cuda环境安装是否成功nvcc --version安装成功后如下图所示六、Python环境搭建1版本要求python3.10Ubuntu22.04默认的版本就是python3.102依赖安装前面通过git已经获取到rknn3-toolkit和rknn3-model-zoo仓库代码。cd rknn3-model-zoo pip install -r requirements.txt export PYTHONPATH./cd rknn3-toolkit/rknn3-toolkit/packages pip install rknn3_toolkit-1.0.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl至此模型转换环境搭建完成
