Skip to content

Commit

Permalink
correct nvidia gpu docs
Browse files Browse the repository at this point in the history
nvidia-cuda-toolkit package is not needed on the host.
  • Loading branch information
andy108369 committed Jun 5, 2024
1 parent 8bfa890 commit fa4d63d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -170,15 +170,15 @@ The `apt install <driver-name>`alternatively allows the install of preferred dri
ubuntu-drivers autoinstall
```

### Install the NVIDIA CUDA Toolkit
### Install the NVIDIA Container Toolkit

```
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | tee /etc/apt/sources.list.d/libnvidia-container.list
apt-get update
apt-get install -y nvidia-cuda-toolkit nvidia-container-toolkit nvidia-container-runtime
apt-get install -y nvidia-container-toolkit nvidia-container-runtime
```

### Additional References for Node Configurations
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ The `apt install <driver-name>`alternatively allows the install of preferred dri
ubuntu-drivers autoinstall
```

### Install the NVIDIA CUDA Toolkit
### Install the NVIDIA Container Toolkit

> _**NOTE**_ - The steps in this sub-section should be completed on all Kubernetes nodes hosting GPU resources
Expand All @@ -95,7 +95,7 @@ curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | tee /etc/apt/sources.list.d/libnvidia-container.list
apt-get update
apt-get install -y nvidia-cuda-toolkit nvidia-container-toolkit nvidia-container-runtime
apt-get install -y nvidia-container-toolkit nvidia-container-runtime
```

### For non-PCIe, e.g. SXM\* GPUs
Expand Down

0 comments on commit fa4d63d

Please sign in to comment.