Merge 9d97e30b99 into ae05379cc9
This commit is contained in:
commit
92cd62e8c9
|
|
@ -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
|
||||
|
||||
|
|
|
|||
|
|
@ -1,2 +1,2 @@
|
|||
torch==2.1.0
|
||||
torchvision==0.16.0
|
||||
torch==2.8.0
|
||||
torchvision==0.23.0
|
||||
|
|
|
|||
Loading…
Reference in New Issue