diff --git a/Dockerfile b/Dockerfile index eed8e9c..5c81240 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM kernai/refinery-parent-images:v2.6.0-torch-cpu +FROM kernai/refinery-parent-images:v3.0.0-torch-cpu WORKDIR /program diff --git a/dev.Dockerfile b/dev.Dockerfile index 97d4315..7f7abaa 100644 --- a/dev.Dockerfile +++ b/dev.Dockerfile @@ -1,4 +1,4 @@ -FROM kernai/refinery-parent-images:v2.6.0-torch-cpu +FROM kernai/refinery-parent-images:v3.0.0-torch-cpu WORKDIR /app diff --git a/gpu.Dockerfile b/gpu.Dockerfile index a2fdf79..a3d5423 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -1,4 +1,4 @@ -FROM kernai/refinery-parent-images:v2.6.0-torch-cuda +FROM kernai/refinery-parent-images:v3.0.0-torch-cuda WORKDIR /program