423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-09-18 12:42:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

update demo docker version to PyTorch 2.5.1

Browse Source
This commit is contained in:
Ronghang Hu 2024-12-11 17:23:19 +00:00
parent ef8e6a9f03
commit 9e6839f6f3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend.Dockerfile
View File

@ -1,4 +1,4 @@
ARG BASE_IMAGE=pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
ARG BASE_IMAGE=pytorch/pytorch:2.5.1-cuda12.1-cudnn9-runtime
ARG MODEL_SIZE=base_plus
FROM ${BASE_IMAGE}