复现Detic详细步骤;

本文详细描述了如何在Ubuntu20.04系统上配置CUDA11.1环境,包括安装CUDNN和相关PyTorch版本,以及克隆并设置Detectron2和Detic项目,以便进行深度学习模型的开发和演示。
摘要由CSDN通过智能技术生成

系统:本人使用系统为Ubuntu20.04,显卡驱动为525(对应cuda12.0)

1.配置cuda环境:cuda11.1(CUDA Toolkit 11.1 Update 1 Downloads) + 对应的cudnn(Download cuDNN v8.0.5 (November gth, 2020), for CUDA 11.1),pytorch版本如下:

pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111

2.

mkdir  ws    (创建一个工作空间)

3. 

git clone github.com:facebookresearch/detectron2.git

4.

cd detectron2

5.

pip install -e .   (体现第1步的重要性)

6.

cd ..

7.

git clone https://github.com/facebookresearch/Detic.git --recurse-submodules (下载另外两个子模块)

8.

cd Detic

9.

pip install -r requirements.txt

10.

pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111

11.设置demo.py中的参数

12.运行demo.py即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值