fix: cuda tag (#949)

Signed-off-by: Keming <kemingyang@tensorchord.ai> Signed-off-by: Keming <kemingyang@tensorchord.ai>
tensorchord · Sep 28, 2022 · d16a975 · d16a975
1 parent 0a4c272
commit d16a975
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/base-images/remote-cache/build.envd b/base-images/remote-cache/build.envd
@@ -3,4 +3,4 @@ def build():
 
 def build_gpu():
     base(os="ubuntu20.04", language="python3")
-    install.cuda(version="11.2", cudnn="8")
+    install.cuda(version="11.2.0", cudnn="8")
diff --git a/pkg/lang/ir/system.go b/pkg/lang/ir/system.go
@@ -93,7 +93,7 @@ func (g Graph) compileCopy(root llb.State) llb.State {
 
 func (g *Graph) compileCUDAPackages(org string) llb.State {
 	return g.preparePythonBase(llb.Image(fmt.Sprintf(
-		"docker.io/%s/%s-cudnn%s-devel-%s",
+		"docker.io/%s:%s-cudnn%s-devel-%s",
 		org, *g.CUDA, *g.CUDNN, g.OS)))
 }