Skip to content

update

update #82

name: AMD PyTorch Job
on:
workflow_dispatch:
pull_request:
push:
jobs:
train:
runs-on: [amdgpu-mi250-x86-64]
steps:
- name: Setup Python
uses: actions/setup-python@v4
with:
python-version: '3.10' # or your preferred version
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install numpy
pip install torch --index-url https://download.pytorch.org/whl/rocm6.2
- name: Create and run training script
run: |
python train.py > training.log 2>&1
- name: Upload logs
uses: actions/upload-artifact@v3
if: always() # Upload logs whether the job succeeds or fails
with:
name: training-logs
path: training.log