idr_torch

Description

Permet de récupérer les variables SLURM afin de gérér le multi-GPU/multi-noeuds pour le parallélisme sur Pytorch.

import torch.distributed as dist
import idr_torch

if idr_torch.rank == 0:
    print(">>> Training on ", len(idr_torch.nodelist), " nodes and ", idr_torch.world_size, " processes") 

dist.init_process_group(backend='nccl', 
                        init_method='env://', 
                        world_size=idr_torch.world_size, 
                        rank=idr_torch.rank)

Si on veut rajouter une nouvelle API, on peut la mettre dans le dossier api et l'importer dans le __init__.py. Ou alors on peut la coder n'importe où, et après appeler idr_torch.register_api(nouvelle_api). Les nouvelles APIs doivent hériter de idr_torch.API. Si on veut faire en sorte que la MASTER_ADDR et le MASTER_PORT soit mis automatiquement (dans le cas où le lanceur ne le fait pas comme SLURM), alors il faut utiliser idr_torch.AutoMasterAddressPort comme décorateur de notre nouvelle API.

On patche aussi le profiler. Il suffit de remplacer from torch.profiler import ... par from idr_torch.profiler import ....

Note : idr_torch est compatible avec submitit.

Metadata

Release files for idr-torch 2.5.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for idr-torch 2.5.0
File	Size	Uploaded	Provenance
idr_torch-2.5.0.tar.gz	11.6 kB	Oct 23, 2025	With provenance	Details

Built distribution (wheel)

Table of built distributions (wheels) for idr-torch 2.5.0
File	Interpreter	ABI	Platform	Reset
idr_torch-2.5.0-py3-none-any.whl 14.7 kB Oct 23, 2025 With provenance	Python 3	none	any	Details

Total release size: 26.4 kB

Release files / idr_torch-2.5.0.tar.gz

Download URL	idr_torch-2.5.0.tar.gz
Size	11.6 kB
Tags	Source
SHA-256 checksum How to use checksums	`e0947b4e3fea03369c7c2a8db362012da889c50c1962c700212f7fcde43998e3`
BLAKE2b-256 checksum How to use checksums	`fa06b888c2c998eb811d9542c72e0fde71effb4fb8c28768f636db060c61f985`
Upload date	Oct 23, 2025
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/6.1.0 CPython/3.13.7`

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/ncassereau/idr_torch
Publishing commit: github.com/ncassereau/idr_torch/tree/87d77bf9c21f10aeca988df34f70c90530828aef

Workflow

Publishing configuration: github.com/ncassereau/idr_torch/blob/87d77bf9c21f10aeca988df34f70c90530828aef/.github/workflows/release.yml
Publishing logs: github.com/ncassereau/idr_torch/actions/runs/18744744673/attempts/1

Artifact

Subject: idr_torch-2.5.0.tar.gz
SHA-256 checksum: e0947b4e3fea03369c7c2a8db362012da889c50c1962c700212f7fcde43998e3 Verifying attestations

Release files / idr_torch-2.5.0-py3-none-any.whl

Download URL	idr_torch-2.5.0-py3-none-any.whl
Size	14.7 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`cba313766c7f93c17905350e009b7d14d2f08e145cf0662b676f3eec4cb51d9a`
BLAKE2b-256 checksum How to use checksums	`d64dc1f4b5a27c41255cb70d853b91c243926f94cc0bf0c6e52fbb3e6e6df91c`
Upload date	Oct 23, 2025
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/6.1.0 CPython/3.13.7`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/ncassereau/idr_torch
Publishing commit: github.com/ncassereau/idr_torch/tree/87d77bf9c21f10aeca988df34f70c90530828aef

Workflow

Publishing configuration: github.com/ncassereau/idr_torch/blob/87d77bf9c21f10aeca988df34f70c90530828aef/.github/workflows/release.yml
Publishing logs: github.com/ncassereau/idr_torch/actions/runs/18744744673/attempts/1

Artifact

Subject: idr_torch-2.5.0-py3-none-any.whl
SHA-256 checksum: cba313766c7f93c17905350e009b7d14d2f08e145cf0662b676f3eec4cb51d9a Verifying attestations

Security for idr-torch 2.5.0

Report project as malware

Provenance

Full provenance

All 2 files in this release have been uploaded with provenance metadata.

Attestation for idr_torch-2.5.0.tar.gz

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/ncassereau/idr_torch
Publishing commit: github.com/ncassereau/idr_torch/tree/87d77bf9c21f10aeca988df34f70c90530828aef

Workflow

Publishing configuration: github.com/ncassereau/idr_torch/blob/87d77bf9c21f10aeca988df34f70c90530828aef/.github/workflows/release.yml
Publishing logs: github.com/ncassereau/idr_torch/actions/runs/18744744673/attempts/1

Artifact

Subject: idr_torch-2.5.0.tar.gz
SHA-256 checksum: e0947b4e3fea03369c7c2a8db362012da889c50c1962c700212f7fcde43998e3 Verifying attestations

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.