diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index cc883da95e..dbecd26e2b 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -17,6 +17,7 @@
 import re
 import shutil
 import subprocess
+import sys
 from distutils.cmd import Command
 from pathlib import Path
 from subprocess import CalledProcessError
@@ -356,7 +357,8 @@ def _compile_embedded_lib(self):
             "-output",
             "feast/embedded_go/lib",
             "-vm",
-            "python3",
+            # Path of current python executable
+            sys.executable,
             "-no-make",
             "github.com/feast-dev/feast/go/embedded"
         ], env={