windows安装PyTorch

Detectron2 是一个强大的目标检测和实例分割框架，由 Facebook AI Research 开发。它基于 PyTorch 框架，并提供了许多先进的算法和模型，如 Faster R-CNN、Mask R-CNN 和 RetinaNet。

本文将介绍如何在 Windows 上安装 Detectron2。

安装步骤

1.安装 Python

Detectron2 需要 Python 3.7 或更高版本，推荐使用Anaconda来管理Python环境。使用conda 来创建新的虚拟环境 ，可以使用以下命令

conda create -n <环境名> python=<Python版本>

执行  conda create -n detectron2 python=3.12.7,即可创建虚拟环境 detectron2 。创建完成后需要激活该环境  ，需要执行命令 conda activate <环境名>

2.安装 PyTorch

Detectron2 基于 PyTorch 框架，因此需要先安装 PyTorch。首先登陆PyTorch的官网 https://pytorch.org/

 我们这里选择windows平台，package 选择Conda或者Pip。CUDA和CPU根据自己电脑情况选择，然后复制命令，粘贴到Anaconda Prompt 或者Anaconda PowerShell Prompt 中，记得先激活对应的虚拟环境。

 3.下载Detectron2 源代码