SO-ARM100/101 安装 Lerobot 环境

来自Waveshare Wiki
跳转至: 导航搜索

安装 Lerobot 环境

对于 Jetson Orin Nano

需要根据你的 CUDA 版本安装 pytorch 和 torchvision 等环境。

1. 安装 Miniconda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh
chmod +x Miniconda3-latest-Linux-aarch64.sh
./Miniconda3-latest-Linux-aarch64.sh
source ~/.bashrc
  • 关闭终端重新打开,确保 conda 环境生效。


2. 创建并激活一个新的 conda 环境用于 lerobot

conda create -y -n lerobot python=3.10 && conda activate lerobot


3. 克隆 LeRobot 仓库:

git clone https://github.com/huggingface/lerobot.git ~/lerobot


4. 使用 miniconda 时,在环境中安装 ffmpeg:

conda install ffmpeg -c conda-forge
  • 若安装后出现编码器问题,显式安装指定版本:
conda install ffmpeg=7.1.1 -c conda-forge


5. 安装带有 feetech 电机依赖的 lerobot:

cd ~/lerobot && pip install -e ".[feetech]"
  • 若下载过程中总是因为连接超时而中断,可以临时指定镜像源来安装:
cd ~/lerobot && pip install -e ".[feetech]" -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com


6. 安装 PyTorch 与 torchvision

  • 根据你的 CUDA 版本安装 pytorch 和 torchvision。可通过以下命令查看 CUDA 版本:
nvcc -V
  • Pytorch 官网查看安装指令,以 CUDA 12.6为例,可以使用以下安装命令:
pip install torch==2.9.0 torchvision==0.24.0 torchaudio==2.9.0 --index-url https://download.pytorch.org/whl/cu126
  • 验证安装:需要在 python 中进行检查
python
import torch
print(torch.cuda.is_available())

输出 True 为正常(有 GPU 支持)。执行 exit() 退出 Python 命令环境。
SO-ARM10X-verify CUDA acceleration.png


7. 安装 OpenCV 依赖

conda install -y -c conda-forge "opencv>=4.10.0.84"  # 通过 conda 安装 OpenCV 和其他依赖,仅适用于 Jetson Jetpack 6.0+
conda remove opencv  # 卸载 OpenCV
pip3 install opencv-python==4.10.0.84  # 使用 pip3 安装指定版本 OpenCV


SO-ARM100/101 教程目录