0
点赞
收藏
分享

微信扫一扫

太速科技-基于6 U VPX M.2 高带宽加固存储板

雨鸣静声 2024-04-27 阅读 14

1. DINO简介

2. DINO环境配置

该部分推荐大家参看DINO源代码中点击进入的配置过程,一般只要一步步来,不会出错的

根据上图大致流程为:

3. 环境配置过程中的细节

3.1 编译过程中遇到CUDA out of memory

当您顺利进行到CUDA算子编译步骤时,会发现有时候显存不够大,可能会报一个CUDA out of memory的错误,这个是我们的显存不够大,造成的,不过编译已经通过了。

RuntimeError: CUDA out of memory. Tried to allocate 1.88 GiB (GPU 0; 11.77 GiB total capacity; 9.41 GiB already allocated; 1.48 GiB free; 9.66 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.  See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

在您的项目目录 DINO-main/models/dino/ops 看到是否编译好的包和库,如下图所示。

3.2 pip install -r requirement.txt过程中 unable to access

这个主要是由于COCO API和Panopticapi 由于有github链接问题,可参看我的往期博文点击进入icon-default.png?t=N7T8https://blog.csdn.net/qq_29750461/article/details/138223735?spm=1001.2014.3001.5502

以上是项目环境配置过程中可能存在的问题,希望帮到大家。

举报

相关推荐

0 条评论