一、安装torch
way1 安装torch-cpu
conda create -n dectectron2py39_d python=3.9
activate dectectron2py39_d
conda install -n dectectron2py39_d cudatoolkit=11.3
conda install -n dectectron2py39_d cudnn=8.2
conda install git
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.2 -c pytorch -c conda-forge
way2 安装torch-gpu
conda create -n dectectron2py39 python=3.9.16 cudnn=8.4.1.50 cudatoolkit=11.3.1
conda activate dectectron2py39
linux+win:pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 way3 网速不行,可能需要离线 离线方法:https://blog.csdn.net/qq_18236971/article/details/131042637
#验证是否成功
import torch
torch.cuda.is_available()
conda install git
二、安装VS2022
安装VS2022社区版,勾选C++桌面开发
三、安装相关离线包
下载离线包 主要参考:https://blog.csdn.net/zzy153/article/details/120693321 (1) 安装cocoapi 点击https://github.com/philferriere/cocoapi下载,然后执行以下指令,另外还需要先安装Cython
pip install Cython -i https://pypi.tuna.tsinghua.edu.cn/simple
cd coco/PythonAPI
python setup.py build_ext --inplace
python setup.py build_ext install
(2) 安装fvcore 点击https://github.com/facebookresearch/fvcore下载,切换到setup.py所在目录,然后执行指令
python setup.py build --force develop
(3) 安装ninja
conda install ninja
四、安装detectron2
安装detectron2 下载https://github.com/facebookresearch/detectron2
python setup.py build develop
五 运行
安装成功
推荐文章
发表评论