Alternative packer setup for gpu

ziglana · Nov 15, 2022 · eb84bfc · eb84bfc
1 parent a55104e
commit eb84bfc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/setup-gpu-node-devices.sh b/setup-gpu-node-devices.sh
@@ -1,7 +1,5 @@
 #!/bin/bash
 
-set -euo pipefail
-
 if /sbin/modprobe nvidia; then
   # Count the number of NVIDIA controllers found.
   NVDEVS=$(lspci | grep -i NVIDIA)

diff --git a/setup-gpu-node.sh b/setup-gpu-node.sh
@@ -11,6 +11,7 @@ bash "${DIR}/setup-node.sh"
 pushd /tmp
 wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
 sh cuda_11.8.0_520.61.05_linux.run --silent --driver
+rm cuda_11.8.0_520.61.05_linux.run
 popd
 
 echo -e "blacklist nouveau\noptions nouveau modeset=0\n" > /etc/modprobe.d/blacklist-nouveau.conf