【学习经验分享NO.22】PyTorch环境搭建

       YOLO系列算法检测需要搭建PyTorch环境才能运行项目，本博客将手把手教学如何搭建PyTorch环境，以YOLOv5检测算法为例，进行讲解。有问题请关注后私信留言，可远程教学。

一、 安装Anaconda、Cuda、pycharm

1.安装Anaconda。根据自己的系统，到官网下载并安装对应版本的的Anaconda。安装链接以及成功 界面如图所示。

Anaconda | The World's Most Popular Data Science Platformhttps://www.anaconda.com/

2.安装Cuda。根据自己的显卡型号，下载对应的的程序。

win+R打开cmd，输入nvidia-smi，下图为博主电脑配置情况

3.安装pycharm。根据自己电脑的配置下载并安装I对应版本。也有其他的IDE，比如VScode等，但本人习惯使用该IDE软件。
PyCharm: the Python IDE for Professional Developers by JetBrainshttps://www.jetbrains.com/pycharm/

二、安装PyTorch

1.打开ananconda prompt

创建环境

# 创建了名叫pytorch，python版本为3.7的新环境
conda create -n pytorch python=3.7  

2.安装PyTorch

以下为PyTorch官网

Start Locally | PyTorchhttps://pytorch.org/get-started/locally/根据自己的操作系统以及Cuda环境等获取安装命令。

获取上面的命令后，打开Anaconda后进行安装，安装过程会比较慢，需要保持网络顺畅，如果期间中断了一次不成功，可以重复进行安装。

# 打开Anaconda Prompt激活名为pytorch的环境
activate pytorch
pip install torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
pip install torchvision-0.8.2+cu110-cp37-cp37m-win_amd64.whl

三、下载YOLOv5项目

GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

 可以看到YOLOv5已经出到7.0的版本，可以选择不同的版本进行下载，其他版本的算法同样可以在Github上搜索后下载获取，然后最好在英文目录下面用Pycharm编辑器打开，将环境配置成已经安装好的PyTorch下。

接着安装YOLOv5所需要的库。需要的库基本都在requiremengts.txt里面有，等待安装成功即可。

pip install requirements.txt