Skip to content

Commit

Permalink
Update setup-gpu-node.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
partouf authored Feb 1, 2025
1 parent f0dde86 commit 9e898de
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup-gpu-node.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,4 +16,4 @@ apt-get -y update
# The CUDA packages include the most appropriate driver for its version.
# Note the hyphen in the version number, not a period.
CUDA_VERSION=12-8
apt install -y nvidia-headless-570-open nvidia-driver-assistant cuda-compiler-${CUDA_VERSION} cuda-runtime-${CUDA_VERSION}
apt install -y nvidia-headless-570-open nvidia-utils-570 nvidia-driver-assistant cuda-compiler-${CUDA_VERSION} cuda-runtime-${CUDA_VERSION}

0 comments on commit 9e898de

Please sign in to comment.