25 lines
825 B
Bash
Executable File
25 lines
825 B
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
echo "Downloading NVIDIA GPG key..."
|
|
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
|
|
|
|
echo "Adding NVIDIA package repository..."
|
|
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
|
|
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
|
|
tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
|
|
|
|
echo "Updating apt..."
|
|
apt-get update
|
|
|
|
echo "Installing NVIDIA Container Toolkit..."
|
|
apt-get install -y nvidia-container-toolkit
|
|
|
|
echo "Configuring Docker..."
|
|
nvidia-ctk runtime configure --runtime=docker
|
|
|
|
echo "Restarting Docker..."
|
|
systemctl restart docker
|
|
|
|
echo "Done! NVIDIA Container Toolkit is installed."
|