Windows11(Cuda12.6.3)的Pytorch+Yolov5安裝與使用

筆者使用Window11，Nvidia RTX 3080Ti顯卡。

Step1)安裝YOLOV5(https://github.com/ultralytics/yolov5)

#cd c:\

#git clone https://github.com/ultralytics/yolov5

Step2)安裝Python虛擬環境

#conda create -n yolov5 python=3.10

#conda activate yolov5

Step3)安裝cuda12.6.3與cudnn9.8.0。

請參考Windows11升級Cuda12.6、Cudnn9.8與安裝yolov4

筆者使用RTX 3080Ti，系統path新增「C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin」與「C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\libnvpp」，系統CUDA_PATH修改為「C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6」，如下圖。

此時執行nvcc -V會顯示12.6，表示cuda為12.6版本。

執行nvidia-smi會顯示驅動程式版本為12.6，目前cuda使用12.6。

Step4)安裝pytorch，到網站https://pytorch.org/get-started/previous-versions/，查詢想要安裝的pytorch版本，筆者安裝pytorch的cuda12.6版本，最接近cuda11.7。

#pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124

Step5)安裝yolov5所需套件

#cd c:\yolov5

#pip install  -r requirements.txt

Step6)開啟攝影鏡頭進行辨識，左邊為攝影機的輸出結果，右邊為即時的辨識結果

#python detect.py --source 0