diff --git a/gpu/compute.go b/gpu/compute.go
index e41a11ac..fcf1910d 100644
--- a/gpu/compute.go
+++ b/gpu/compute.go
@@ -1501,6 +1501,7 @@ func (g *compute) Release() {
 	for _, a := range g.atlases {
 		a.Release()
 	}
+	g.ctx.Release()
 	*g = compute{}
 }