Skip to content

Commit

Permalink
test lighter image
Browse files Browse the repository at this point in the history
  • Loading branch information
Mikel Broström committed Jul 9, 2024
1 parent 52865c7 commit 83f2afd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# docker build -t mikel-brostrom/yolov5_strongsort_osnet .

# Base image: Nvidia PyTorch https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
FROM nvcr.io/nvidia/pytorch:24.06-py3
FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime

# Update image
RUN apt update
Expand Down

0 comments on commit 83f2afd

Please sign in to comment.