步骤参考:目标检测---教你利用yolov5训练自己的目标检测模型_didiaopao的博客-CSDN博客
Q1: 在使用“conda install pytorch torchvision torchaudio cudatoolkit=11.1命令安装pytorch时,若出现错误:“PackagesNotFoundError: The following packages are not available from current channels: - cudatoolkit=11.1”
A1: 可先输入conda config --add channels conda-forge 命令后再进行安装。
Q2:
正常的结果: 失败的结果:
A2: 如果有独立显卡应该安装GPU版本的,但是可能之前电脑上安装过CPU版本的pytorch,现在想安装GPU版本的,从官网复制命令后,下载的还是CPU版本的pytorch
最后发现原因是安装过CPU版本的pytorch后,会给电脑安装一个名字叫做cpuonly的模块,要换成GPU,需要卸载这个模块conda uninstall cpuonly,卸载时好像会自动替换为GPU版本的pytorch。
Q3: