File size: 498 Bytes
5b17847 9316d0f 5b17847 9316d0f | 1 2 3 4 5 6 7 8 9 | #!/bin/bash
set -e
echo "=== job start: $(date) ==="
nvidia-smi || true
pip install -q --no-cache-dir -r /code/code/requirements.txt
echo "=== deps installed: $(date) ==="
python -c "import torch, torchvision, transformers, datasets, accelerate, pycocotools, huggingface_hub; print('torch', torch.__version__, 'cuda', torch.cuda.is_available(), 'tf', torchvision.__version__, 'transformers', transformers.__version__, 'datasets', datasets.__version__)"
exec python -u /code/code/train_detr.py "$@"
|