yolov5使用cpu推理

Pytorch安装(cpu版)
安装cpu版本的pytorch不需要安装cuda,安装过python之后直接安装cpu版本的pytorch即可运行
cpu/torch下载地址
ctrl+f 搜索如cpu/torch-1.8
再根据python版本及系统选择,这里选的是python3.8 对应的1.8.0版本的pytorch
在这里插入图片描述
再根据下载的pytorch版本下载对应的torchvison
yolov5使用cpu推理_第1张图片
还在刚才的链接地址搜索cpu/torchvision 选择对应torch及python版本的文件,如
在这里插入图片描述
切换到下载文件所在路径,指定python版本安装

$ python3.8 -m pip install --user cpu/torch-1.8          //双击tab补全文件名

再安装torchvision

$ python3.8 -m pip install --user cpu/torchvision-0.9          //补全文件名

你可能感兴趣的:(pytorch,python,深度学习)