Skip to content

Commit

Permalink
cpu offline
Browse files Browse the repository at this point in the history
  • Loading branch information
anarkiwi committed Oct 3, 2024
1 parent 8229d80 commit 3e11fb3
Show file tree
Hide file tree
Showing 2 changed files with 73 additions and 13 deletions.
73 changes: 73 additions & 0 deletions offline-cuda.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
---
# VOL_PREFIX=/data RECORDING=/logs/fm.sigmf-meta docker compose -f offline.yml up gamutrf
networks:
gamutrf:
driver: bridge
driver_opts:
com.docker.network.driver.mtu: 9000
default:
external: true
name: none
services:
mqtt:
restart: no
image: iqtlabs/edgetech-mqtt-dev:v1.0
networks:
- gamutrf
ports:
- '1883:1883'
gamutrf:
restart: no
image: iqtlabs/gamutrf:v0.5.71
privileged: true
networks:
- gamutrf
ports:
- '9001:9001'
- '10000:10000'
- '10001:10001'
- '10002:10002'
cap_add:
- SYS_NICE
- SYS_RAWIO
devices:
- /dev/bus/usb:/dev/bus/usb
- /dev/dri/renderD128:/dev/dri/renderD128
# nvidia container toolkit etc must be installed.
# Comment out "deploy" if on non-x86 platform (e.g. Pi4)
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['0']
# count: 1
capabilities: [gpu]
volumes:
- '${VOL_PREFIX}:/logs'
- '/tmp/.X11-unix:/tmp/.X11-unix'
environment:
- 'HOME=/logs'
- 'XDG_RUNTIME_DIR=/tmp'
- 'NVIDIA_DRIVER_CAPABILITIES=all'
- 'NVIDIA_VISIBLE_DEVICES=all'
- DISPLAY
command:
- gamutrf-offline
- '${RECORDING}'
- --nfft=1024
- --tune-step-fft=512
- --tuneoverlap=1
- --bucket_range=1
- --pretune
- --sample_dir=/logs/samples
- --write_samples=100000000
- --iq_inference_model_server=torchserve:8080
- --iq_inference_model_name=torchsig_model
- --no-iq_inference_background
- --no-vkfft
healthcheck:
test: [CMD, "/gamutrf/bin/scanhc.sh", "9000"]
interval: 10s
timeout: 10s
retries: 3
13 changes: 0 additions & 13 deletions offline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,19 +30,6 @@ services:
cap_add:
- SYS_NICE
- SYS_RAWIO
devices:
- /dev/bus/usb:/dev/bus/usb
- /dev/dri/renderD128:/dev/dri/renderD128
# nvidia container toolkit etc must be installed.
# Comment out "deploy" if on non-x86 platform (e.g. Pi4)
deploy:
resources:
reservations:
devices:
- driver: nvidia
device_ids: ['0']
# count: 1
capabilities: [gpu]
volumes:
- '${VOL_PREFIX}:/logs'
- '/tmp/.X11-unix:/tmp/.X11-unix'
Expand Down

0 comments on commit 3e11fb3

Please sign in to comment.