2025年6月18日 星期三

python程式調用GPU的第一步:安裝 cuda + cudnn

實在不懂 nvdia 到底為什麼不把安裝 cuda + cudnn 的流程弄簡便一點 每次安裝都要檢查 顯示卡、python、cuda、pytorch 各種版本的相容性 學學人家 windows 的 .net framework 搞成一鍵安裝 
 記錄一下詢問 chatgpt 的內容 

1.安裝cuda (找合適版本)
這次 ChatGPT 建議用 11.8 ,我就下載 cuda 11.8 


2.安裝cudnn
cudnn 這邊就沒有特別去挑版本

3.安裝 pytorch
pytorch 這邊就記得選好 cuda 的版本,再複製「Run this Command」的內容
去做 pip install 


=================================
 這次是為了kibo 競賽要做物件偵測 yolo 模型微調