trying PyPantograph install aesop

This commit is contained in:
Brando Miranda 2024-06-03 20:17:58 -07:00
parent e6ac8d3a8a
commit 45a2ec127b
2 changed files with 28 additions and 0 deletions

1
PyPantograph Symbolic link
View File

@ -0,0 +1 @@
/afs/cs.stanford.edu/u/brando9/PyPantograph

27
main_krbtmux_bm.sh Normal file
View File

@ -0,0 +1,27 @@
#!/bin/bash
# SNAP cluster specific main sh script to run jobs
# - snap: https://ilwiki.stanford.edu/doku.php?id=snap-servers:snap-servers and support il-action@cs.stanford.edu
# - live server stats: https://ilwiki.stanford.edu/doku.php?id=snap-servers:snap-gpu-servers-stats
# source $AFS/.bashrc.lfs
source $AFS/.bash_profile
conda activate pypantograph_env
export CUDA_VISIBLE_DEVICES=1
export CUDA_VISIBLE_DEVICES=$(nvidia-smi --query-gpu=memory.used --format=csv,nounits,noheader | awk '{print NR-1 " " $1}' | sort -nk2 | head -n1 | cut -d' ' -f1)
echo CUDA_VISIBLE_DEVICES = $CUDA_VISIBLE_DEVICES
# -- Run
# - Pull merge with PyPantograph main
cd $HOME/PyPantograph
# git checkout brando
git pull origin main
git submodule update --init
conda activate pypantograph_env
poetry build
pip install -e $HOME/PyPantograph
pip install -e $HOME/gold-ai-olympiad
python $HOME/PyPantograph/pantograph/server.py
# -- Demo
# python ~/snap-cluster-setup/src/train/simple_train_train.py
# CUDA_VISIBLE_DEVICES=5 python src/test_vllm.py