基于pytorch的yolov5的环境配置

1.安装anaconda

进入官网进行下载安装Anaconda | The Operating System for AI

安装详细步骤 

根据自己的操作系统是32位还是64位选择对应的版本到官网下载，但是官网下载龟速，建议到清华大学镜像站下载，https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Windows-x86_64.exe

然后开始安装

 最后点击finish，结束安装

打开anaconda promote输入

conda --version

若出现conda 版本号则是安装成功 

2.利用conda下载pytorch

（1）创建环境

打开anaconda promote，进入命令行输入以下代码

conda create –n yolov5 python=3.10

我的python版本是3.10,创建了一个yolov5的环境

y/n，选择y进行下载安装

安装成功后输入以下指令，检查是否安装成功

conda env list #检查当前拥有的环境

（2） 进入虚拟环境

conda activate yolov5

在虚拟环境中安装pytorch（主要需要安装pytorch, torchvision,torchaudio三个包 ），安装torch之前需要查看自己显卡驱动的版本。

根据自己的电脑配置选择合适的CUDA版本。
（1）首先确定自己的显卡型号，确定算力
（2）确定自己的CUDA Driver 版本，查看方式：命令行输入

nvidia-smi

由此可见我的cuda版本是12.2，官网只更新到12.1，便直接下载了12.1，意外发现还是匹配

（3）下载pytorch 

在已经打开的虚拟环境中，利用pip进行安装。

在下面网址选择与自己cuda相对应的torch版本进行下载：

Previous PyTorch Versions | PyTorch

 以我为例：

pip  --trusted-host http://pypi.tuna.tsinghua.edu.cn install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121

由于官方下载，由于国外源下载速度较慢，我这里选择了国内的清华源镜像只需要在pip后面添加以下代码即可正常下载：

--trusted-host http://pypi.tuna.tsinghua.edu.cn

根据提示输入y进行pytorch的安装 。

（4）验证

进入到yolov5的环境中，输入以下指令：

conda list

出现这三个包说明gpu版本的pytorch已经下载完成了