Skip to content

Instantly share code, notes, and snippets.

View bashkanov's full-sized avatar

Oleksii Bashkanov bashkanov

  • Germany / Ukraine
View GitHub Profile
@bashkanov
bashkanov / agents_to_gpus.sh
Created February 26, 2021 15:07
Utility scripts for distributing wandb agents on multiple GPUs.
#!/usr/bin/env bash
function usage() {
echo "Usage: bash $0 -v /home/env --gpus 3 --agent \"wandb agent __agent__\""
echo "Automatically populates agents in separate screens and GPUs."
echo ""
echo "Arguments:"
echo -e " -v --venv\t\tPath to the virtual enviroment with wandb and tensorflow installed."
echo -e " -a --agent\t\tAgent command"
echo -e " -g --gpus\t\tGPUs to use. Can be comma-separated like: 0,1"