bug: fix missing function in example mnist (#589)

* why it disappeared? Signed-off-by: Jinjing.Zhou <allenzhou@tensorchord.ai> * fix Signed-off-by: Jinjing.Zhou <allenzhou@tensorchord.ai>
tensorchord · Jul 11, 2022 · 54dfc52 · 54dfc52
1 parent 67b4e5d
commit 54dfc52
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/mnist/build.envd b/examples/mnist/build.envd
@@ -10,3 +10,7 @@ def build():
     ])
     shell("zsh")
     config.jupyter(password="", port=8888)
+
+def build_gpu():
+    build()
+    install.cuda(version="11.6", cudnn="8")