This commit is contained in:
jason-e-gross 2025-09-25 18:46:33 +00:00 committed by GitHub
commit 92cd62e8c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions

View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
FROM nvidia/cuda:12.8.1-base-ubuntu22.04
ENV DEBIAN_FRONTEND noninteractive
ENV CMDARGS --listen

View File

@ -1,2 +1,2 @@
torch==2.1.0
torchvision==0.16.0
torch==2.8.0
torchvision==0.23.0