Update dockers/base-cuda/Dockerfile

wangraying · akihironitta · web-flow · commit 167d0f81380e · 2022-01-19T00:57:53.000+08:00
Co-authored-by: Aki Nitta &lt;nitta@akihironitta.com&gt;
diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile
@@ -117,7 +117,7 @@ RUN \
 
 RUN \
     # install Bagua
-    pip install bagua-cuda${BAGUA_CUDA_VERSION}==0.9.0
+    pip install bagua-cuda${CUDA_VERSION//.}==0.9.0
 
 RUN \
     # Show what we have
