fix: update setup script to install PaddlePaddle GPU version from official source
Changes to setup_dev_env.sh: - Add support for CUDA 13.x (install CUDA 12.x compatible version) - Use official PaddlePaddle source for GPU versions - Install paddlepaddle-gpu==3.0.0b2 from official index - CUDA 13.x: use cu123 package (backward compatible) - CUDA 12.x: use cu123 package - CUDA 11.7+: use cu118 package - CUDA 11.2-11.6: use cu117 package Changes to requirements.txt: - Comment out paddlepaddle dependency - Let setup script handle GPU/CPU version installation This fixes the issue where pip installed CPU-only paddlepaddle 3.2.1 instead of GPU version, causing GPU acceleration to be unavailable. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -106,6 +106,9 @@ echo -e "${YELLOW}[6/9] 偵測 GPU 和 CUDA 支援...${NC}"
|
||||
|
||||
# GPU 偵測函數
|
||||
detect_gpu() {
|
||||
# 初始化變量
|
||||
PADDLE_INDEX=""
|
||||
|
||||
# 檢查是否有 NVIDIA GPU
|
||||
if command -v nvidia-smi &> /dev/null; then
|
||||
echo -e "${GREEN}✓ 偵測到 NVIDIA GPU${NC}"
|
||||
@@ -120,19 +123,29 @@ detect_gpu() {
|
||||
CUDA_MAJOR=$(echo $CUDA_VERSION | cut -d. -f1)
|
||||
CUDA_MINOR=$(echo $CUDA_VERSION | cut -d. -f2)
|
||||
|
||||
if [ "$CUDA_MAJOR" -ge 12 ]; then
|
||||
if [ "$CUDA_MAJOR" -ge 13 ]; then
|
||||
echo -e "${YELLOW}⚠ CUDA 13.x 偵測到${NC}"
|
||||
echo "PaddlePaddle 目前最高支援 CUDA 12.x"
|
||||
echo "將嘗試安裝 CUDA 12.x 編譯的 GPU 版本(可能兼容)"
|
||||
USE_GPU=true
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu==3.0.0b2" # 使用支援 CUDA 12.x 的版本
|
||||
PADDLE_INDEX="https://www.paddlepaddle.org.cn/packages/stable/cu123/"
|
||||
elif [ "$CUDA_MAJOR" -eq 12 ]; then
|
||||
echo "將安裝 PaddlePaddle GPU 版本 (CUDA 12.x)"
|
||||
USE_GPU=true
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu"
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu==3.0.0b2"
|
||||
PADDLE_INDEX="https://www.paddlepaddle.org.cn/packages/stable/cu123/"
|
||||
elif [ "$CUDA_MAJOR" -eq 11 ]; then
|
||||
if [ "$CUDA_MINOR" -ge 7 ]; then
|
||||
echo "將安裝 PaddlePaddle GPU 版本 (CUDA 11.7+)"
|
||||
USE_GPU=true
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu"
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu==3.0.0b2"
|
||||
PADDLE_INDEX="https://www.paddlepaddle.org.cn/packages/stable/cu118/"
|
||||
elif [ "$CUDA_MINOR" -ge 2 ]; then
|
||||
echo "將安裝 PaddlePaddle GPU 版本 (CUDA 11.2-11.6)"
|
||||
USE_GPU=true
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu"
|
||||
PADDLE_PACKAGE="paddlepaddle-gpu==3.0.0b2"
|
||||
PADDLE_INDEX="https://www.paddlepaddle.org.cn/packages/stable/cu117/"
|
||||
else
|
||||
echo -e "${YELLOW}⚠ CUDA 版本過舊 ($CUDA_VERSION),建議升級到 11.2+${NC}"
|
||||
echo "將安裝 CPU 版本"
|
||||
@@ -172,10 +185,16 @@ echo ""
|
||||
echo -e "${YELLOW}安裝 PaddlePaddle...${NC}"
|
||||
if [ "$USE_GPU" = true ]; then
|
||||
echo "安裝 GPU 加速版本..."
|
||||
pip install $PADDLE_PACKAGE -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||||
if [ -n "$PADDLE_INDEX" ]; then
|
||||
echo "使用官方源: $PADDLE_INDEX"
|
||||
pip install "$PADDLE_PACKAGE" -i "$PADDLE_INDEX"
|
||||
else
|
||||
echo "使用 PyPI..."
|
||||
pip install "$PADDLE_PACKAGE"
|
||||
fi
|
||||
else
|
||||
echo "安裝 CPU 版本..."
|
||||
pip install $PADDLE_PACKAGE -i https://pypi.tuna.tsinghua.edu.cn/simple
|
||||
pip install paddlepaddle
|
||||
fi
|
||||
|
||||
# 安裝其他依賴(跳過 requirements.txt 中的 paddlepaddle)
|
||||
|
||||
Reference in New Issue
Block a user