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的原因是电脑的虚拟内存不是系统自动分配的,默认等于系统内存,一般你自己修改过就容易出问题。
越来越远的告别
浙公网安备 33010602011771号