chore(deps): cap torch ceiling at <2.12, pin Linux wheels to cu128 (#3570)

* chore(deps): ceiling + cuda

* ci: bump cuda version docker image

* ci: add cpu wheel to release workflow

* chore(deps): update uv.lock

* docs: update installation with cuda note
This commit is contained in:
Steven Palma
2026-05-11 19:47:55 +02:00
committed by GitHub
parent 6d269b28c8
commit 26ff40ddd7
6 changed files with 327 additions and 142 deletions
+1 -1
View File
@@ -18,7 +18,7 @@
# docker build -f docker/Dockerfile.internal -t lerobot-internal .
# Configure the base image for CI with GPU access
ARG CUDA_VERSION=12.6.3
ARG CUDA_VERSION=12.8.1
ARG OS_VERSION=24.04
FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu${OS_VERSION}