فهرست منبع

feat(zhch): 更新GPU安装说明,确保驱动和CUDA版本匹配,优化安装步骤

zhch158_admin 3 ماه پیش
والد
کامیت
b84a8bca95
1فایلهای تغییر یافته به همراه19 افزوده شده و 8 حذف شده
  1. 19 8
      zhch/study-notes.md

+ 19 - 8
zhch/study-notes.md

@@ -22,16 +22,27 @@ paddlex --install PaddleOCR PaddleDetection PaddleClas  # 例如PaddleOCR
 # uv pip install paddleocr==3.1.0
 ```
 
-# linux GPU
-```
-# GPU 版本,需显卡驱动程序版本 ≥550.54.14(Linux)或 ≥550.54.14(Windows)
-uv pip install paddlepaddle-gpu==3.1.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
+# linux GPU - 更新版本以匹配NVML 570.169
+```bash
+# 首先检查NVIDIA驱动版本
+nvidia-smi
+
+# 确保驱动版本 ≥ 570.169
+# 如果需要更新驱动
+sudo apt update
+sudo apt install nvidia-driver-570  # 或更新版本
+
+# 安装匹配的CUDA版本(推荐12.6以匹配您的驱动)
+wget https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/cuda_12.6.0_560.28.03_linux.run
+sudo sh cuda_12.6.0_560.28.03_linux.run
+
+# 安装PaddlePaddle GPU版本
+uv pip install -U --reinstall paddlepaddle-gpu==3.1.1 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
 uv pip install -e .
-paddlex --install PaddleOCR PaddleDetection PaddleClas  # 例如PaddleOCR
-# uv pip install paddleocr==3.1.0
+paddlex --install PaddleOCR PaddleDetection PaddleClas
 
-uv pip install --upgrade pybind11
-uv pip install --upgrade faiss-cpu
+# 验证安装
+python -c "import paddle; print(f'CUDA available: {paddle.device.is_compiled_with_cuda()}'); print(f'GPU count: {paddle.device.cuda.device_count()}')"
 ```
 
 # 设置launch.json