Skip to content

chore: add isGPUWorker label to cellposesam_train - #158

Merged
arjunrajlab merged 1 commit into
masterfrom
claude/cellposesam-train-gpu-label
Jul 27, 2026
Merged

chore: add isGPUWorker label to cellposesam_train#158
arjunrajlab merged 1 commit into
masterfrom
claude/cellposesam-train-gpu-label

Conversation

@arjunrajlab

Copy link
Copy Markdown
Collaborator

Summary

Adds the missing isGPUWorker="true" Docker label to the Cellpose-SAM retrain worker (cellposesam_train).

The worker is already GPU-accelerated — it uses a CUDA base image, declares com.nvidia.volumes.needed, and calls cellpose.core.use_gpu() at runtime — but its LABEL block was missing the isGPUWorker flag that the platform uses to identify GPU workers. This brings it in line with the main cellposesam worker and cellpose_train.

Change

 LABEL isUPennContrastWorker="" \
+      isGPUWorker="true" \
       isAnnotationWorker="" \

No code or runtime-behavior changes — Docker metadata only.

🤖 Generated with Claude Code

The Cellpose-SAM retrain worker is GPU-accelerated (CUDA base image,
com.nvidia.volumes.needed, cellpose.core.use_gpu at runtime) but was
missing the isGPUWorker="true" Docker label. Add it to match the main
cellposesam worker and cellpose_train.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@arjunrajlab
arjunrajlab merged commit 34d0057 into master Jul 27, 2026
1 check passed
@arjunrajlab
arjunrajlab deleted the claude/cellposesam-train-gpu-label branch July 27, 2026 00:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant