YOLOv5配置环境
前言
在使用YOLOv5前我们首先要配置它的相关环境,网上有很多教程,这里我放一个我学的
教程链接
一、下载相关IDE
这里推荐大家下载Anaconda
我自己使用的是pycharm,在使用YOLO的时候感觉比anaconda复杂了不少。
虽然pycharm也有jupyter的插件,但是还是推荐大家安装Anaconda可以省去安装很多第三方库的步骤。
非要和我一样头铁使用pychram也行就是麻烦了些,记得在pycharm中的库安装更改成镜像源就行。
二、pytroch下载与安装
首先取pytorch的官网,官网会根据你的电脑推荐你下载对应的pytorch版本。
最重要的是会告诉你应该安装什么版本的CUDA
这里你别着急取下载,放着这个页面,先去下载CUDA和cudnn。
PS: 我这里使用的是pip安装,因为我是用的pycharm
如果你用的是Anaconda这里选择Conda安装
还有CUDA10.2 pytorch 已经不支持了,所以一般都是CUDA11.3,我们在下载的时候就选择CUDA11.3就好了
三、CUDA和cudnn下载
这里直接根据上边教程取下载就行,不赘述。
后续
根据推荐的教程进行。