TOFIX: configure docker

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

tests/e2e/e2e_test.go

@@ -61,10 +61,6 @@ func TestMain(t *testing.T) {
 }
 
 var _ = BeforeSuite(func() {
-	var (
-		err error
-	)
-
 	getTestEnv()
 
 	runner = NewRunner(
@@ -91,7 +87,13 @@ var _ = BeforeSuite(func() {
 	Expect(err).ToNot(HaveOccurred())
 
 	if installCTK {
-		_, _, err = toolkitInstaller.Install(runner)
+		_, _, err := toolkitInstaller.Install(runner)
+		Expect(err).ToNot(HaveOccurred())
+
+		_, _, err = runner.Run(`sudo nvidia-ctk runtime configure --runtime=docker`)
+		Expect(err).ToNot(HaveOccurred())
+
+		_, _, err = runner.Run(`sudo systemctl restart docker`)
 		Expect(err).ToNot(HaveOccurred())
 	}
 })