version 0.3.0

.gitignore

@@ -1,3 +1,4 @@
 testdata/*-dfp.tiff
 testdata/*-ffp.tiff
 .python-version
+*.sw*

Dockerfile

@@ -12,7 +12,7 @@ ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:
 # Installing necessary packages
 
 # Installing basicpy and other pip packages
-RUN pip --no-cache-dir install basicpy==1.0.1
+RUN pip --no-cache-dir install git+https://github.com/peng-lab/BaSiCpy@dev
 
 # Copy script and test run
 COPY ./main.py /opt/

build.sh

@@ -1,4 +1,6 @@
 VERSION=0.3.0
+
+docker buildx create --use
 docker buildx build \
     --push \
     --build-arg BASE_IMAGE="yfukai/conda-jax:latest" \

jax_image/Dockerfile

@@ -2,7 +2,7 @@ ARG MINIFORGE_NAME="Miniforge3"
 ARG MINIFORGE_VERSION="22.9.0-3"
 ARG BASE_IMAGE="ubuntu:20.04"
 ARG JAX_VERSION_EXTRA="cpu"
-ARG JAX_VERSION="0.3.22"
+ARG JAX_VERSION="0.4.23"
 
 FROM ${BASE_IMAGE}
 #https://stackoverflow.com/questions/44438637/arg-substitution-in-run-command-not-working-for-dockerfile

jax_image/build.sh

@@ -1,5 +1,6 @@
 JAX_VERSION=0.4.23
 
+docker buildx create --use
 docker buildx build \
     --push \
     --platform linux/amd64,linux/arm64/v8 \
@@ -11,9 +12,8 @@ docker buildx build \
 docker buildx build \
     --push \
     --platform linux/amd64 \
-    --build-arg BASE_IMAGE="nvidia/cuda:11.5.2-cudnn8-devel-ubuntu20.04" \
     --build-arg JAX_VERSION="$JAX_VERSION" \
-    --build-arg JAX_VERSION_EXTRA="cuda11_cudnn82" \
+    --build-arg JAX_VERSION_EXTRA="cuda12_pip" \
     -t yfukai/conda-jax:$JAX_VERSION-cuda \
     -t yfukai/conda-jax:latest-cuda \
     "."