1️⃣ 打開 Anaconda Prompt / CMD

請在 Anaconda Prompt命令提示字元 (CMD) 中執行以下步驟(不要在 Python 交互模式 >>> 下執行!)。

2️⃣ 激活 yolov5-env 環境

如果你已經有 yolov5-env 環境,執行:

sh
conda activate yolov5-env

如果還沒有這個環境,請先創建:

sh
conda create --name yolov5-env python=3.10 -y conda activate yolov5-env

3️⃣ 安裝 PyTorch

YOLOv5 需要 PyTorch,你可以選擇安裝 CUDA(NVIDIA GPU)或 CPU 版本:

✅ GPU 版本(NVIDIA CUDA)

如果你的電腦有 NVIDIA 顯卡,請安裝 PyTorch(CUDA 11.8 版):

sh
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

✅ CPU 版本(適用於沒有 NVIDIA 顯卡的電腦)

如果你 沒有 NVIDIA 顯卡,請安裝 CPU 版本:

sh
pip install torch torchvision torchaudio

4️⃣ 下載 YOLOv5

請執行:

sh
git clone https://github.com/ultralytics/yolov5.git cd yolov5

如果沒有安裝 git,請先安裝:

sh
conda install git -y

5️⃣ 安裝 YOLOv5 依賴

執行:

sh
pip install -r requirements.txt

🔹 測試 YOLOv5 是否安裝成功

你可以運行 YOLOv5 內建的測試命令來檢查安裝是否成功。

sh
python detect.py --weights yolov5s.pt --img 640 --source data/images/zidane.jpg
  • 如果你的顯示卡有 CUDA 支援,但它沒有被使用,請加上 --device 0 來強制使用 GPU:

    sh
    python detect.py --weights yolov5s.pt --img 640 --source data/images/zidane.jpg --device 0
  • 如果要用 CPU 運行,請加上 --device cpu

    sh
    python detect.py --weights yolov5s.pt --img 640 --source data/images/zidane.jpg --device cpu

如果成功,會在 runs/detect/exp 目錄下看到偵測結果!🚀


🔹 常見問題

1️⃣ "ModuleNotFoundError: No module named 'torch'"

這表示 PyTorch 沒有安裝成功,請重新執行:

sh
pip install torch torchvision torchaudio

或者如果你要用 CUDA(NVIDIA GPU),執行:

sh
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

2️⃣ "git: command not found"

請安裝 git

sh
conda install git -y

3️⃣ "RuntimeError: No CUDA GPUs are available"

如果你有 NVIDIA 顯卡,但 YOLOv5 無法使用 GPU,請確認:

  • 你是否安裝了對應的 NVIDIA 驅動程式

  • 你是否安裝了對應的 CUDA 版本

  • 嘗試使用 --device 0 強制使用 GPU:

    sh
    python detect.py --weights yolov5s.pt --img 640 --source data/images/zidane.jpg --device 0

這樣你應該就能成功安裝 YOLOv5 了!🎯

創作者介紹
創作者 IMM 的頭像
IMM-AT

IMM

IMM-AT 發表在 痞客邦 留言(0) 人氣( 0 )