Skip to content

Commit

Permalink
Build gamutRF image with bespoke torchsig libraries, for new augmenta…
Browse files Browse the repository at this point in the history
…ton script.
  • Loading branch information
anarkiwi committed Nov 6, 2023
1 parent 042cf38 commit 7fdfed3
Show file tree
Hide file tree
Showing 3 changed files with 24 additions and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/docker-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ jobs:
docker build -f Dockerfile.base . -t iqtlabs/gamutrf-base:latest
cd ..
docker build -f Dockerfile . -t iqtlabs/gamutrf:latest
docker build -f docker/Dockerfile.torchsig docker -t iqtlabs/gamutrf-torchsig:latest
docker run -t iqtlabs/gamutrf:latest gamutrf-scan --help
docker run -t iqtlabs/gamutrf:latest gamutrf-sigfinder --help
docker run -t iqtlabs/gamutrf:latest gamutrf-api --help
Expand Down
9 changes: 9 additions & 0 deletions .github/workflows/docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -84,3 +84,12 @@ jobs:
push: true
tags: iqtlabs/gamutrf:${{ steps.change_version.outputs.VERSION }}
if: github.repository == 'iqtlabs/gamutrf' && github.event_name == 'push'
- name: Build and push gamutrf-torchsig
uses: docker/build-push-action@v5
with:
context: docker
file: docker/Dockerfile.torchsig
platforms: linux/amd64,linux/arm64
push: true
tags: iqtlabs/gamutrf-torchsig:${{ steps.change_version.outputs.VERSION }}
if: github.repository == 'iqtlabs/gamutrf' && github.event_name == 'push'
14 changes: 14 additions & 0 deletions docker/Dockerfile.torchsig
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
FROM ubuntu:22.04 as torchsig-builder
WORKDIR /root
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends build-essential ca-certificates cmake git python3-pip python3-dev && pip install -U pip
RUN pip install git+https://github.com/TorchDSP/[email protected]

FROM iqtlabs/gamutrf:latest
WORKDIR /root
ENV DEBIAN_FRONTEND noninteractive
# TODO: find a better way cherrypick just Torchsig itself, without Torch et al.
COPY --from=torchsig-builder /usr/local/lib/python3.10/dist-packages/torchsig /usr/local/lib/python3.10/dist-packages/torchsig
RUN apt-get update && apt-get install -y --no-install-recommends python3-pip && pip3 install PyWavelets numba torch --extra-index-url https://download.pytorch.org/whl/cpu && apt-get purge -y python3-pip
RUN python3 -c "from torchsig.transforms import transforms"
RUN python3 -c "from gamutrf import grscan"

0 comments on commit 7fdfed3

Please sign in to comment.