依照pytorch 的安裝說明 的話,cuda or cpu only各有明確的指令。
例如:
pip 安裝, cuda support:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113-- cpu 的話,就 cu113 改 cpu。
conda 安裝:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch-- cpu 的話就是 cudatoolkit=11.3 改成 cpuonly
但是看 yolov5 的 requirements.txt,其中 torch 的部份..:
torch>=1.7.0沒有指定 ..
-- 所以會是有 cuda support ?
用 pip install torch,不加上 url。安裝完後,用 torch.cuda.is_available() 測試看看。
在有 cuda 的機器上,是 True,在沒有 cuda 的機器上,是 False。
但是用 pip freeze 看安裝的 package full name 都是 torch=1.11.0
-- 發現,若是 10.2,則pip install torch 時、不須指定 download whl。反而是 cpu 板才要。
所以 package manager 版本應該有 support 了。
測試,在 cuda 的機器上 pip install torch 指定 download whl cpu,cuda.is_available() 是 False.
結論大概是..
package manager : pip 已經把 pytorch 10.2, 11.3 都納入 package repo 中了。但是 conda 沒有。所以 conda 安裝一定要指定 cudatoolkit 和 python channel
yolov5 其實有要求 python 版本:
Python 3.7.0 required by YOLOv5
沒有留言:
張貼留言