From ff5c3e036a98ecb92e5f7555fffcfe9ba440796f Mon Sep 17 00:00:00 2001
From: joy <muyun688@aliyun.com>
Date: Tue, 2 Dec 2025 16:20:38 +0800
Subject: [PATCH] 123

123
---
 scripts/cuda.sh | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/scripts/cuda.sh b/scripts/cuda.sh
index ea411b1..2c6174a 100644
--- 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