Jetson Orin CUDA 12 workspace

pixi.toml

[workspace]
channels = ["conda-forge", "robostack-jazzy"]
platforms = [{ name = "jetson-cuda12", platform = "linux-aarch64", cuda = "12", glibc = "2.35" }]

[dependencies]
pytorch = ">=2.11.0,<3"
torchvision = ">=0.26.0,<0.27"
lerobot = ">=0.6.0,<0.7"
ros-jazzy-desktop = ">=0.11.0,<0.12"
arm-variant = { build = "*tegra*" }

[tasks.verify]
cmd = "python -c \"import torch, torchvision, lerobot, numpy; print('numpy', numpy.__version__); x=torch.randn(1024,1024,device='cuda'); torch.cuda.synchronize(); print('STACK OK', float((x@x).sum()), '| torch', torch.__version__, '| tv', torchvision.__version__, '| lerobot', lerobot.__version__)\""