docker image is reduced in size
This commit is contained in:
@@ -3,7 +3,7 @@
|
||||
ARG GPU=nvidia
|
||||
|
||||
FROM nvidia/cuda:13.0.2-cudnn-runtime-ubuntu24.04 AS base-nvidia
|
||||
FROM rocm/dev-ubuntu-24.04:7.2.1-complete AS base-amd
|
||||
FROM rocm/dev-ubuntu-24.04:7.2.1 AS base-amd
|
||||
|
||||
FROM base-${GPU} AS app
|
||||
|
||||
|
||||
Reference in New Issue
Block a user