Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

trl env report all cuda devices #2216

Merged
merged 1 commit into from
Oct 11, 2024
Merged

trl env report all cuda devices #2216

merged 1 commit into from
Oct 11, 2024

Conversation

qgallouedec
Copy link
Member

@qgallouedec qgallouedec commented Oct 10, 2024

What does this PR do?

$ trl env

  Copy-paste the following information when reporting an issue:

  - Platform: Linux-5.15.0-1048-aws-x86_64-with-glibc2.31
  - Python version: 3.11.9
  - PyTorch version: 2.4.1
- - CUDA device: NVIDIA H100 80GB HBM3
+ - CUDA device(s): NVIDIA H100 80GB HBM3, NVIDIA H100 80GB HBM3
  - Transformers version: 4.46.0.dev0
  - Accelerate version: 0.34.2
  - Accelerate config: not found
  - Datasets version: 3.0.0
  - HF Hub version: 0.24.7
  - TRL version: 0.12.0.dev0+b3f93f0
  - bitsandbytes version: 0.41.1
  - DeepSpeed version: 0.15.1
  - Diffusers version: 0.30.3
  - Liger-Kernel version: 0.3.0
  - LLM-Blender version: 0.0.2
  - OpenAI version: 1.46.0
  - PEFT version: 0.13.0

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline,
    Pull Request section?
  • Was this discussed/approved via a GitHub issue? Please add a link
    to it if that's the case.
  • Did you make sure to update the documentation with your changes? Here are the
    documentation guidelines.
  • Did you write any new necessary tests?

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@qgallouedec qgallouedec requested review from kashif and lewtun October 10, 2024 20:12
@qgallouedec qgallouedec changed the title trl env report all coda devices trl env report all cuda devices Oct 11, 2024
@qgallouedec qgallouedec merged commit 5375d71 into main Oct 11, 2024
9 of 10 checks passed
@qgallouedec qgallouedec deleted the multi-gpu-trl-env branch October 11, 2024 14:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants