fix

flexflow · Jul 15, 2023 · 51ffe1f · 51ffe1f
1 parent 915cbb9
commit 51ffe1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/run.sh b/docker/run.sh
@@ -28,7 +28,7 @@ fi
 
 
 if [[ "$image" == "flexflow-environment" ]]; then
-    eval docker run -it "$gpu_arg" "--shm-size=${SHM_SIZE}" "flexflow-environment-${FF_GPU_BACKEND}:latest"
+    eval docker run -it "$gpu_arg" "--shm-size=${SHM_SIZE}" "flexflow-environment-${FF_GPU_BACKEND}-11.6.2:latest"
 elif [[ "$image" == "flexflow" ]]; then
     eval docker run -it "$gpu_arg" "--shm-size=${SHM_SIZE}" "flexflow-${FF_GPU_BACKEND}:latest"
 elif [[ "$image" == "mt5" ]]; then

diff --git a/python/flexflow/flexflow_python b/python/flexflow/flexflow_python
@@ -3,7 +3,7 @@
 # This file runs the flexflow_python interpreter when installing FlexFlow via pip. This file can be ignored when installing FlexFlow using cmake/make
 
 python_packages=$(python -c "from distutils import sysconfig; print(sysconfig.get_python_lib(plat_specific=False,standard_lib=False))")
-pylib_path="$("$python_packages"/flexflow/findpylib.py)"
+pylib_path="$(python "$python_packages"/flexflow/findpylib.py)"
 pylib_dir="$(dirname "$pylib_path")"
 export PATH="${python_packages}/flexflow/bin:${PATH}"
 export LD_LIBRARY_PATH="${python_packages}/flexflow/lib:${pylib_dir}:${PATH}"