Files
PHANTOM/scripts/setuptpu.sh

10 lines
191 B
Bash

commands = (
"pip install \"jax[tpu]\" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html"
"pip install stable-baselines3>=2.2.0 gymnasium wandb tensorboard"
"
)