Win10 YoloV5学习笔记

1. 没有英伟达显卡的测试环境下,可以使用CPU进行AI训练和推理。
conda install pytorch torchvision torchaudio cpuonly -c pytorch,如果给conda指定了国内镜像源,那么不能有官网的-c pytorch指令。
2. 通过conda安装torch、torchvision的时候可以安装cudatoolkit,而pip安装的torch则需要单独去安装cuda。
3. 确保torch、torchvision、cuda为同一版本,包括py版本和cuda版本号。
4. 训练或者推理时候提示分页文件错误或者cv:out of meomry的原因是电脑的虚拟内存不是系统自动分配的,默认等于系统内存,一般你自己修改过就容易出问题。
 
posted @ 2022-07-07 16:33  Tracyrvp  阅读(125)  评论(0)    收藏  举报