123

2025-12-02 16:20:38 +08:00 · 2025-12-02 16:20:38 +08:00 · ff5c3e036a
parent f226f95345
commit ff5c3e036a
1 changed files with 4 additions and 2 deletions
--- a/scripts/cuda.sh
+++ b/scripts/cuda.sh
@ -160,8 +160,10 @@ EOF
    fi
    
    # 生效环境变量
-    export LC_BYOBU=0 && source "/etc/profile" #临时解决
-    
+    export PATH=${CUDA_INSTALL_DIR}/bin:\$PATH
+    #export LC_BYOBU=0 && source "/etc/profile" #临时解决
+    export LC_BYOBU=0
+    export LD_LIBRARY_PATH=${CUDA_INSTALL_DIR}/lib64:\$LD_LIBRARY_PATH
    # 验证安装
    step "验证 CUDA 安装"
    if command -v nvcc &>/dev/null; then