Skip to content

Commit e76b8ca

Browse files
committed
replace nvidia-docker with docker --gpus=all
1 parent 444393b commit e76b8ca

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

README.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@ Requirements
1616
Linux
1717
~~~~~
1818

19-
* For ``sdist`` and CUDA (x86_64) wheel build, run the tool on Linux (x86_64). ``nvidia-docker`` and NVIDIA GPU are required for Verify step.
20-
* For CUDA (JetPack (aarch64)) wheel build, run the tool on Linux (x86_64 with QEMU (aarch64) or Tegra). ``nvidia-docker`` is required for Verify step.
19+
* For ``sdist`` and CUDA (x86_64) wheel build, run the tool on Linux (x86_64). NVIDIA GPU + NVIDIA Container Toolkit are required for Verify step.
20+
* For CUDA (aarch64) wheel build, run the tool on Linux (x86_64 with QEMU (aarch64), aarch64, or Tegra). NVIDIA GPU + NVIDIA Container Toolkit is required for Verify step.
2121
* For ROCm wheel build, run the tool on Linux (x86_64). AMD GPU is required for Verify step.
2222

2323
Notes:

dist.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,7 @@ def _run_container(
267267
image_tag, kind))
268268
docker_run = ['docker', 'run']
269269
if kind == 'cuda' and require_runtime:
270-
docker_run = ['nvidia-docker', 'run']
270+
docker_run += ['--gpus=all']
271271
elif kind == 'rocm':
272272
targets = os.environ.get('HCC_AMDGPU_TARGET', None)
273273
if targets is None:

0 commit comments

Comments
 (0)