From 51ffe1fa12295f5d38996f69fc0d23790a786d71 Mon Sep 17 00:00:00 2001 From: Gabriele Oliaro Date: Fri, 14 Jul 2023 22:38:14 -0400 Subject: [PATCH] fix --- docker/run.sh | 2 +- python/flexflow/flexflow_python | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/run.sh b/docker/run.sh index e04e7d68c1..182af511a1 100755 --- 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 index 471a0d07fb..7fed992c6d 100644 --- 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}"