Permalink
Find file
Fetching contributors…
Cannot retrieve contributors at this time
24 lines (21 sloc) 712 Bytes
diff --git a/python/tensorboard/tensorboard b/python/tensorboard/tensorboard
index cde1e9e..e205d52 100755
--- a/python/tensorboard/tensorboard
+++ b/python/tensorboard/tensorboard
@@ -7,6 +7,7 @@ import shutil
import sys
import subprocess
import zipfile
+import site
# Return True if running on Windows
def IsWindows():
@@ -58,6 +59,10 @@ def FindModuleSpace():
module_space = stub_filename + '.runfiles'
if os.path.isdir(module_space):
break
+ for mod in site.getsitepackages():
+ module_space = mod + '/tensorboard/tensorboard' + '.runfiles'
+ if os.path.isdir(module_space):
+ return module_space
runfiles_pattern = "(.*\.runfiles)/.*"
if IsWindows():