YOLOv8的GPU环境搭建方法
首先说明这个环境搭建教程是基于电脑已经安装好CUDA和CUDNN的情况下,去搭建能够正确运行YOLOv8代码的Pytorch的GPU版本。具体安装方法可见:最适合新手入门的CUDA、CUDNN、Pytorch安装教程_cuda安装-CSDN博客
第一步:需要在cmd中创建虚拟环境conda create -n 环境名字 python版本,例如创建一个名字为YOLOv8的虚拟环境,并且指定Python版本为3.9,则需要在cmd中输入的代码为:
conda create -n YOLOv8 python==3.9
即可成功创建。
第二步:激活指定的虚拟空间,例如激活名为YOLOv8的虚拟环境,其代码如下:
conda activate YOLOv8
就会进入对应的虚拟环境。
第三步:在虚拟环境中安装运行YOLOv8官方代码所需要的模块。推荐首先安装ultralytics模块。如果下载较慢,可以添加镜像源。使用镜像源安装模块的方法可见:Python使用镜像安装模块_python安装镜像-CSDN博客
pip install ultralytics
第四步:安装对应的CUDA的Pytorch版本,首先进入Pytorch官网,选择对应的版本进行pip安装,如下图:
但是在这里存在一个问题,安装很慢很慢,往往超时,我们可以将安装命令输入CMD中,会产生如下图:
复制此下载路径到迅雷中,粘贴到新建下载任务中,然后下载完成后,激活虚拟环境,然后进入到下载此文件的路径下,输入如下代码:
pip install “torch-2.4.1+cu118-cp39-cp39-win_amd64.whl”
即可安装成功,但是在运行YOLOv8代码的时候,会报错torchvision与CUDA不适配,主要是在安装ultralytics模块的时候安装的torchvision是cpu版本的,torch + torchaudio都是coda版本,torchvision却不是,这就是问题所在,需要手动卸载torchvision,然后输入对应的cuda版本的torchvision安装,实例代码如下:
pip install torchvision==0.19.1 --index-url https://download.pytorch.org/whl/cu118
参考博客:[YOLOv8] 问题分析和解决 - NotImplementedError: Could not run ‘torchvision::nms‘ ...from the ‘CUDA‘ backend_yolov8notimplementederror: could not run 'torchvis-CSDN博客
完美解决下载安装pytorch cuda版本超级慢的问题_pytorch下载太慢-CSDN博客