Wait for networking before loading nvidia modules (#1962)

Co-authored-by: Nikolay Kvetsinski <kvetsins@amazon.com>
awslabs · Sep 10, 2024 · fcab352 · fcab352
1 parent c6ea158
commit fcab352
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/templates/al2023/runtime/gpu/nvidia-kmod-load.service b/templates/al2023/runtime/gpu/nvidia-kmod-load.service
@@ -1,9 +1,9 @@
 [Unit]
 Description=Loading NVIDIA kernel modules
-# load modules after cloud-init
-After=cloud-final.service
-Requires=cloud-final.service
-Before=nvidia-persistenced.service
+# the script needs to use IMDS, so wait for the network to be up to avoid any flakiness
+After=network-online.target
+Wants=network-online.target
+Before=nvidia-fabricmanager.service nvidia-persistenced.service
 
 [Service]
 Type=oneshot