gen_shell_env.py

import os.path as osp
import sys
import subprocess as sp
from pathlib import Path

PROJ_DIR = osp.abspath(osp.dirname(__file__))
print(PROJ_DIR)
PROJ_NAME = 'TAG-Benchmark'
LINUX_HOME = str(Path.home())
CONDA_ENV_NAME = 'ct'
CONDA_PATH = f'{LINUX_HOME}/miniconda/envs/{CONDA_ENV_NAME}'
HTOP_FILE = f"{CONDA_PATH}/bin/nvidia-htop.py"
SV_INIT_CMDS = [
    f'source {LINUX_HOME}/miniconda/etc/profile.d/conda.sh;conda activate {CONDA_ENV_NAME}',
]

env_vars = {
    # PATHS
    'LP': PROJ_DIR,  # Local Path
    'PROJ_NAME': PROJ_NAME,  # PROJ Name
    'HTOP_FILE': HTOP_FILE
}

server_setting_file = f'{PROJ_DIR}/shell_env.sh'
with open(server_setting_file, 'w') as f:
    for var_name, var_val in env_vars.items():
        f.write(f'export {var_name}="{var_val}"\n')

    for cmd in SV_INIT_CMDS:
        f.write(f'{cmd}\n')
    print()