Skip to content

Commit 935f180

Browse files
author
Yi Sun
committed
remove nvidia cuda
1 parent 4161dc3 commit 935f180

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

cellpose/4.0.7/Dockerfile

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
1+
FROM ubuntu:22.04
22

3-
LABEL base_image="nvidia/cuda:11.8.0-runtime-ubuntu22.04"
3+
LABEL base_image="ubuntu:22.04"
44
LABEL version="1"
55
LABEL software="cellpose"
66
LABEL software.version="4.0.7"
@@ -80,5 +80,4 @@ RUN pip install --no-cache-dir -U pip \
8080
tqdm \
8181
tifffile \
8282
fastremap \
83-
torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --extra-index-url https://download.pytorch.org/whl/cu118 \
8483
'cellpose[gui]'==$CELLPOSE_VERSION

0 commit comments

Comments
 (0)