如果结果为+CPU 就说明这个torch版本是CPU版本不支持显卡训练,如果你就是想GPU训练就需要从新安装,对应torch的版本选择,Yolov5官方给的是≥ 1.8 如果先要跟换GPU版本可以去下方链接中寻找 download.pytorch.org/whl/torch_stable.html cu121/torch-2.1.0%2Bcu121-cp39-cp39-win_amd64:其中的cu121对应的CUDA版本...
官网链接:https://pytorch.org/get-started/locally/根据自己的需求选择,如图所示,我们用cuda11.1为例 复制图中我标出的代码 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge 1. -2.打开cmd,激活你为yolov5配置的环境 activate 你的环境名 1. -3.下载pytorch 输入上...
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 三、环境创建 创建运行环境 conda create -n yolov5_cuda10.2 python=3.9 输入Yes下载依赖 完成 其中“yolov5_cuda10.2” 是虚拟环境的名字,.conda\envs中有虚拟环境数据和依赖 激活环境 conda activate yolov5_cuda...