Final commit

docker/Dockerfile

@@ -1,11 +1,14 @@
 # If you want to use CUDA for visibility subtraction (much faster!), you need
 # to adjust this number to the NVIDIA driver on the host system.
-ARG NVIDIA_VERSION=11.4.3
+# Or you can supply e.g. --build-arg NVIDIA_VERSION=12.2.0 to the docker build
+# command.
+ARG NVIDIA_VERSION
+# ARG NVIDIA_VERSION=12.2.0
 
 # CPU
-FROM ubuntu:22.04
+# FROM ubuntu:22.04
 # CUDA version
-# FROM nvidia/cuda:${NVIDIA_VERSION}-devel-ubuntu20.04
+FROM nvidia/cuda:${NVIDIA_VERSION}-devel-ubuntu20.04
 
 ENV DEBIAN_FRONTEND="noninteractive"
 RUN apt-get update -y && \
@@ -37,14 +40,15 @@ ENV RUSTUP_HOME=/opt/rust CARGO_HOME=/opt/cargo PATH=/opt/cargo/bin:$PATH
 ENV RUST_VERSION=1.72
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --profile minimal --default-toolchain=$RUST_VERSION
 
-# CPU
-RUN git clone https://github.com/MWATelescope/mwa_hyperdrive -b sdc3 \
-  && /opt/cargo/bin/cargo install --path mwa_hyperdrive --locked --features=cfitsio-static,hdf5-static \
-  && rm -r mwa_hyperdrive
-# # CUDA
-# RUN git clone https://github.com/MWATelescope/mwa_hyperdrive -b sdc3 \
-#   && /opt/cargo/bin/cargo install --path mwa_hyperdrive --locked --features=cfitsio-static,hdf5-static,cuda \
+# # CPU
+# RUN git clone https://github.com/MWATelescope/mwa_hyperdrive -b SDC3 \
+#   && /opt/cargo/bin/cargo install --path mwa_hyperdrive --locked --features=cfitsio-static,hdf5-static \
 #   && rm -r mwa_hyperdrive
+# CUDA
+# Add the gpu-single feature to make CUDA code run faster but with less precision (should not seriously affect results)
+RUN git clone https://github.com/MWATelescope/mwa_hyperdrive -b SDC3 \
+  && /opt/cargo/bin/cargo install --path mwa_hyperdrive --locked --features=cfitsio-static,hdf5-static,cuda \
+  && rm -r mwa_hyperdrive
 
 RUN git clone https://github.com/cjordan/sdc3_vis_convert \
   && /opt/cargo/bin/cargo install --path sdc3_vis_convert --locked \
@@ -55,3 +59,6 @@ RUN cd /chips \
   && ls \
   && make \
   && cd /
+
+ADD sdc3.toml /sdc3.toml
+ADD sdc3_inner_lobes+trecs_rot-0.15_10.fits /sdc3_inner_lobes+trecs_rot-0.15_10.fits

docker/sdc3.toml

@@ -0,0 +1,19 @@
+[data]
+array_position = [116.764, -26.825, 0.0] # Discovered in OSKAR logs
+# array_position = [116.7644482, -26.670051, 0.0] # Jack's coords; these appear to produce worse results
+dut1 = -0.9504120730644577 # Determined with scipy.optimize based off of provided station positions and UVWs
+
+[sky-model]
+source_list = "/sdc3_inner_lobes+trecs_rot-0.15_10.fits"
+veto_threshold = 0
+source_dist_cutoff = 90
+#num_sources = 3000
+
+[model]
+no_precession = true
+
+[beam]
+beam_type = "ska_airy"
+
+[vis-subtract]
+invert = true