diff --git a/cmd/nvidia-ctk-installer/toolkit/installer/executables.go b/cmd/nvidia-ctk-installer/toolkit/installer/executables.go index c98971c0..ad5d42bc 100644 --- a/cmd/nvidia-ctk-installer/toolkit/installer/executables.go +++ b/cmd/nvidia-ctk-installer/toolkit/installer/executables.go @@ -141,7 +141,7 @@ func (w *wrapper) Install(destDir string) error { func (w *wrapper) render() (io.Reader, error) { wrapperTemplate := `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" diff --git a/cmd/nvidia-ctk-installer/toolkit/installer/executables_test.go b/cmd/nvidia-ctk-installer/toolkit/installer/executables_test.go index 18e09cad..7cbd2a92 100644 --- a/cmd/nvidia-ctk-installer/toolkit/installer/executables_test.go +++ b/cmd/nvidia-ctk-installer/toolkit/installer/executables_test.go @@ -36,7 +36,7 @@ func TestWrapperRender(t *testing.T) { }, expected: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -55,7 +55,7 @@ fi }, expected: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -73,7 +73,7 @@ PATH=/foo/bar/baz \ }, expected: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" diff --git a/cmd/nvidia-ctk-installer/toolkit/installer/installer_test.go b/cmd/nvidia-ctk-installer/toolkit/installer/installer_test.go index fdeb6298..5932d472 100644 --- a/cmd/nvidia-ctk-installer/toolkit/installer/installer_test.go +++ b/cmd/nvidia-ctk-installer/toolkit/installer/installer_test.go @@ -167,7 +167,7 @@ func TestToolkitInstaller(t *testing.T) { mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -183,7 +183,7 @@ XDG_CONFIG_HOME=/foo/bar/baz/.config \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -199,7 +199,7 @@ XDG_CONFIG_HOME=/foo/bar/baz/.config \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -215,7 +215,7 @@ XDG_CONFIG_HOME=/foo/bar/baz/.config \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -230,7 +230,7 @@ PATH=/foo/bar/baz:$PATH \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -245,7 +245,7 @@ PATH=/foo/bar/baz:$PATH \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@" @@ -261,7 +261,7 @@ PATH=/foo/bar/baz:$PATH \ mode: 0777, wrapper: `#! /bin/sh -cat /proc/modules | grep -e \"^nvidia \" >/dev/null 2>&1 +cat /proc/modules | grep -e "^nvidia " >/dev/null 2>&1 if [ "${?}" != "0" ]; then echo "nvidia driver modules are not yet loaded, invoking runc directly" exec runc "$@"