openVLA 推理镜像
实现openVLA的推理,输入图片,得到动作输出
镜像简介
- 功能: 这个镜像主要用于openVLA推理
- 特点: 预装了openVLA环境,下载了openVLA权重,一键实现推理,清晰学习openVLA的模型结构
环境与依赖
本镜像构建和运行所需的基础环境。
- 框架及版本: (例如:PyTorch 2.2)
- CUDA版本: (例如:CUDA 12.4)
- 其他依赖: (例如:Python 3.10, GCC 7.5)
配置方法
- 具体配置步骤
cd /workspace/openvla-oft 后
直接执行
python openval_predict.py 即可实现模型推理
环境验证代码
直接执行
python openval_predict.py 得到openVLA模型结构输出,根据对应图片输出了
动作,说明运行成功
相关链接
常见问题
此镜像为优云首个 openVLA镜像。
此镜像为 王道计算机教育(B站有70多万粉丝)王道训练营的AI课程的授课镜像之一,对于王道训练营的AI课程感兴趣的同学,可以加微信 cskaoyanbb1了解,也可以直接到B站,或者王道在线 微信公众号,直接联系对应客服了解课程。