Skip to content

pytorch 2.4.1, torchvision 0.19.1 #281

pytorch 2.4.1, torchvision 0.19.1

pytorch 2.4.1, torchvision 0.19.1 #281

Workflow file for this run

name: CI Test
on: [push, pull_request]
jobs:
ci-test:
runs-on: ubuntu-22.04
steps:
- uses: actions/checkout@v4
with:
fetch-depth: 1
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: '3.11'
- name: Install Poetry
uses: snok/install-poetry@v1
with:
virtualenvs-create: true
virtualenvs-in-project: true
version: 1.8.3
- name: Install poetry and dependencies
# TODO: remove workaround for https://github.com/pytorch/pytorch/issues/100974 once torch upgraded.
run: |
sudo apt-get -yq update && sudo apt-get install -yq libnvidia-compute-535-server nvidia-cuda-dev
git submodule init && \
git submodule update && \
sed -i "s/torch==[0-9\.]+/torch/g" torchsig/pyproject.toml && \
poetry config virtualenvs.create true && \
poetry install && \
for i in torch torchvision ; do poetry run pip install --no-cache-dir "$i"=="$(grep $i\\s pyproject.toml | grep -Eo '\"[0-9\.]+' | sed 's/\"//g')" || exit 1 ; done
poetry run python -c 'import torch' && \
poetry run python -c 'import rfml.annotation_utils as annotation_utils' && \
poetry run black . --check --force-exclude torchsig
- name: Run e2e test
run: |
mkdir -p testdata && \
DATA=$(realpath testdata) ./utils/run_siggen.sh && \
rm -rf $(poetry env info -p) && \
DATA=$(realpath testdata) ./utils/gamutrf_offline_siggen.sh