openvla的本地复现步骤

openvla的本地复现步骤

git clone https://gitee.com/caodev/mirror-openvla.git

依赖处理:

conda命令不好使,依赖跟不上5090D

pip install torch==2.9.0+cu130 torchvision==0.24.0+cu130 –index-url https://download.pytorch.org/whl/cu130

pip install torchaudio==2.9.0+cu130 –index-url https://download.pytorch.org/whl/cu130

同步修改 pyproject.toml 的版本,避免被覆盖

问题处理

有一些DNN的警告报错信息

2025-11-05 15:55:55.934310: I tensorflow/core/util/port.cc:113] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable `TF_ENABLE_ONEDNN_OPTS=0`.
2025-11-05 15:55:55.952274: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:9261] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
2025-11-05 15:55:55.952298: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:607] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-11-05 15:55:55.952904: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1515] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2025-11-05 15:55:55.956019: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 AVX512F AVX512_VNNI AVX512_BF16 AVX_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.

一般这些没啥大问题,如果看着难受,给日志屏蔽了。

机器人/openvla/openvla实验/openvla的本地复现步骤.txt · 最后更改: 2025/11/05 07:59
CC Attribution 4.0 International 除额外注明的地方外,本维基上的内容按下列许可协议发布: CC Attribution 4.0 International