diff --git a/setup-gpu-node.sh b/setup-gpu-node.sh index 35832f710..59248f4dc 100644 --- a/setup-gpu-node.sh +++ b/setup-gpu-node.sh @@ -15,5 +15,8 @@ popd 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. +# TODO @rwarmstr suggests using just nvidia-headless--open for just +# the driver, IF we don't need CUDA installed systemwide. @mgodbolt wasn't +# sure what we actually need on the host. CUDA_VERSION=12-8 apt install -y cuda-compiler-${CUDA_VERSION} cuda-runtime-${CUDA_VERSION}