Skip to content

[Proposal] Fixes to install and docker-compose files for running on 50 series GPU #2226

Closed
@LJones-alt

Description

@LJones-alt

Proposal

Current documentation to run IsaacLab on docker is missing setup insructions for running on 50 series.

Motivation

Updates to the docker-compose file will fix some of the main issues users are reporting when running IsaacLab in docker (required for Ubuntu 24). Examples/ tutorials do not work out of the box due to torch / torchvision / torchaudio library conflicts on the 50 series GPUs. The Nvidia nvoptix is also not pulled through correctly in the docker image causing noisy rendering on 50 series GPU.

Alternatives

Add the nightly build for pytorch, torchaudio and torchvision as a not in the docker image setup instructions.

Update docker-compose to include nvoptix binding

- type: bind
  source: /usr/share/nvidia/nvoptix/bin
  target: /usr/share/nvidia/nvoptix/bin
  read_only: true

Add instructions for 50 series GPU to update docker-compose to include the nightly torch, torchvision and torchaudio builds:
update the entry point for the base image :
entrypoint: ["bash", "-c", "pip install --upgrade --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 && bash"]

Additional context

Update this line to include torchaudio and torchvision
Image

Checklist

  • [ x] I have checked that there is no similar issue in the repo (required)

Acceptance Criteria

Add the criteria for which this task is considered done. If not known at issue creation time, you can add this once the issue is assigned.

  • Criteria 1
  • Criteria 2

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions