diff --git a/Docker/install_wrapper.sh b/Docker/install_wrapper.sh
index f3860ffe..6dd93e5a 100644
--- a/Docker/install_wrapper.sh
+++ b/Docker/install_wrapper.sh
@@ -18,7 +18,7 @@ ln -s /workspace/models/pretrained_models /workspace/GPT-SoVITS/GPT_SoVITS/pretr
 
 ln -s /workspace/models/G2PWModel /workspace/GPT-SoVITS/GPT_SoVITS/text/G2PWModel
 
-bash install.sh --device "CU${CUDA_VERSION}" --source HF
+bash install.sh --device "CU${CUDA_VERSION//./}" --source HF
 
 pip cache purge