๋ณธ๋ฌธ ๋ฐ”๋กœ๊ฐ€๊ธฐ
์—๋Ÿฌ ๋ฐ ์ด์Šˆ

Colab MMDetection ์…‹์—… ์ด์Šˆ - Colab์—์„œ ์‚ฌ์šฉx

by kaizen_bh 2025. 12. 10.

 

 

 

์ฝ”๋žฉ์˜ torch, cuda ์‚ฌ์–‘์€ ์•„๋ž˜์™€ ๊ฐ™๋‹ค

import torch
print("torch:", torch.__version__)
print("cuda :", torch.version.cuda)
>>>
torch: 2.9.0+cu126
cuda : 12.6

 

 

MMDetectino์„ ์‹คํ–‰ํ•˜๊ธฐ ์œ„ํ•ด ํ•„์š”ํ•œ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋“ค์„ ์„ค์น˜ํ•ด์ฃผ๋ฉด

!pip install -U openmim
!mim install mmengine

 

Traceback (most recent call last):
  File "/usr/local/bin/mim", line 3, in <module>
    from mim.cli import cli
  File "/usr/local/lib/python3.12/dist-packages/mim/__init__.py", line 10, in <module>
    import setuptools  # noqa: F401
    ^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/dist-packages/setuptools/__init__.py", line 16, in <module>
    import setuptools.version
  File "/usr/local/lib/python3.12/dist-packages/setuptools/version.py", line 1, in <module>
    import pkg_resources
  File "/usr/local/lib/python3.12/dist-packages/pkg_resources/__init__.py", line 2172, in <module>
    register_finder(pkgutil.ImpImporter, find_on_path)
                    ^^^^^^^^^^^^^^^^^^^
AttributeError: module 'pkgutil' has no attribute 'ImpImporter'. Did you mean: 'zipimporter'?

 

์š”๋ž˜ ๋ฉ๋‹ˆ๋‹ค

๊ทธ๋ž˜์„œ ์„œ๋ฒ„ ์…‹ํŒ…ํ•  ๋•Œ torch, cuda ๋ฒ„์ „์„ ๋‚ฎ์ถฐ์„œ ์„ค์น˜ํ•ด์ฃผ๊ณ  ์ง„ํ–‰ํ–ˆ์„ ๋•Œ ์ด์ƒ์—†์—ˆ๋˜ ๋ฐฉ๋ฒ•์„ ์‚ฌ์šฉํ•ด๋ณด์•˜๋‹ค 

 

https://bh-kaizen.tistory.com/59#object-detection-%EB%8C%80%ED%9A%8C-%EC%84%9C%EB%B2%84-%EC%85%8B%EC%97%85

 

uv๋กœ ํ”„๋กœ์ ํŠธ ํ™˜๊ฒฝ ์…‹ํŒ…ํ•˜๊ธฐ

๋„ค์ด๋ฒ„ ๋ถ€์บ ์—์„œ ๋ฆฌ๋ทฐ ๊ฐ์„ฑ ๋ถ„๋ฅ˜ ์ฑŒ๋ฆฐ์ง€๋ฅผ ํ•˜๋Š”๋ฐ venv ๋Œ€์‹  uv๋ฅผ ์‚ฌ์šฉํ•ด๋ณด๋ฉด ๋” ๋น ๋ฅด๊ณ  ์ข‹๋‹ค๊ณ  ํ•ด์„œ ์ด๋ฒˆ ๊ธฐํšŒ์— ์จ๋ณด๋ คํ•œ๋‹ค์…‹ํŒ…ํ•˜๋А๋ผ ํ—ท๊ฐˆ๋ฆฌ๋Š” ๋ถ€๋ถ„๋“ค์ด ์ƒ๊ธฐ๊ธฐ๋„ ํ•ด์„œ ๊ทธ๋ƒฅ ์…‹ํŒ… ๊ณผ์ •์„ ๊ฐ„๋‹จ

bh-kaizen.tistory.com

 

 

 

# 2. PyTorch 2.1.0 + cu121 ์„ค์น˜ (ํ™˜๊ฒฝ ๊ณ ์ •)
!pip install torch==2.1.0 --index-url 'https://download.pytorch.org/whl/cu121'
!pip install torchvision==0.16.0 --index-url 'https://download.pytorch.org/whl/cu121'

# 3. ํ™˜๊ฒฝ ํ™•์ธ

import torch
print("torch:", torch.__version__)
print("cuda :", torch.version.cuda)
>>>
Looking in indexes: https://download.pytorch.org/whl/cu121ERROR: Could not find a version that satisfies the requirement torch==2.1.0 
(from versions: 2.2.0+cu121, 2.2.1+cu121,2.2.2+cu121, 2.3.0+cu121, 2.3.1+cu121, 2.4.0+cu121, 2.4.1+cu121, 2.5.0+cu121, 2.5.1+cu121)
ERROR: No matching distribution found for torch==2.1.0
Looking in indexes: https://download.pytorch.org/whl/cu121
ERROR: Could not find a version that satisfies the requirement torchvision==0.16.0 (from versions: 0.1.6, 0.2.0, 0.17.0+cu121, 0.17.1+cu121, 0.17.2+cu121, 0.18.0+cu121, 0.18.1+cu121, 0.19.0+cu121, 0.19.1+cu121, 0.20.0+cu121, 0.20.1+cu121)
ERROR: No matching distribution found for torchvision==0.16.0

torch: 2.9.0+cu126
cuda : 12.6

 

 

์š”๋ ‡๊ฒŒ ํ•ด๋‹น ๋ฒ„์ ผ์„ ๋ชป์ฐพ๋Š”๋‹ค๋ฉฐ ์„ค์น˜ ์‹คํŒจ

 

 

 

https://github.com/chulminkw/DLCV_New/blob/main/faster_rcnn/kaggle_new/mm_faster_rcnn_train_coco_bccd.ipynb

 

DLCV_New/faster_rcnn/kaggle_new/mm_faster_rcnn_train_coco_bccd.ipynb at main · chulminkw/DLCV_New

Contribute to chulminkw/DLCV_New development by creating an account on GitHub.

github.com

 

 

MMDetection ์„ค์น˜

  • ๊ฐ•์˜ ์˜์ƒ์—๋Š” pip install mmcv-full๋กœ mmcv๋ฅผ ์„ค์น˜(์•ฝ 10๋ถ„ ์ •๋„์˜ ์‹œ๊ฐ„์ด ์†Œ์š”)
  • ์‹ค์Šต์ฝ”๋“œ๋Š” pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.13/index.html ๋กœ ๋ณ€๊ฒฝ(์„ค์น˜์— 12์ดˆ ์ •๋„ ๊ฑธ๋ฆผ. 2022.09).
  • 2023๋…„ 4์›” 6์ผ ๊ธฐ์ค€์œผ๋กœ mmdetection์ด 3.0์œผ๋กœ upgrade๋จ. ์‹ค์Šต ์ฝ”๋“œ๋Š” mmdetection 2.x ๊ธฐ์ค€์œผ๋ฏ€๋กœ mmdetection 2.x ์†Œ์Šค์ฝ”๋“œ ์„ค์น˜ ํ•„์š”.
  • 2024๋…„ 9์›” colab์˜ numpy version์ด 1.24๋กœ upgrade๋˜๋ฉด์„œ ์ผ๋ถ€ ์ฝ”๋“œ๊ฐ€ ๋™์ž‘์˜ค๋ฅ˜. numpy 1.23 ์œผ๋กœ downgrade ์ ์šฉ
  • 2025๋…„ 1์›” 17์ผ Colab์˜ python ๋ฒ„์ „์ด 3.10์—์„œ 3.11๋กœ ๋ฒ„์ „์—… ๋˜๋ฉด์„œ pytorch 2.0, torchvision 0.15๋กœ ๋ณ€๊ฒฝ. mmcv๋„ !pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/index.html ๋กœ ๋ณ€๊ฒฝ.
  • 2025๋…„ 8์›” 25์ผ Colab์˜ python ๋ฒ„์ „์ด 3.11์—์„œ 3.12๋กœ ๋ฒ„์ „์—… ๋˜๋ฉด์„œ ๋” ์ด์ƒ mmcv-full์ด ์ œ๋Œ€๋กœ ์„ค์น˜ ๋˜์ง€ ์•Š์Œ.
  • ์ด์— Colab ํ™˜๊ฒฝ์—์„œ Kaggle ํ™˜๊ฒฝ์œผ๋กœ ์‹ค์Šตํ™˜๊ฒฝ ์ด๊ด€. Kaggle์€ ์—ฌ์ „ํžˆ python ๋ฒ„์ „์ด 3.11์ž„.
  • ๊ธฐ์กด Colab์˜ ๋””๋ ‰ํ† ๋ฆฌ ๊ตฌ์กฐ๋Š” /content ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ ์‹ค์Šต์ฝ”๋“œ ์ˆ˜ํ–‰๋จ. Kaggle์—์„œ๋Š” /kaggle/working์ด๋ฉฐ ์ž๋™์œผ๋กœ ํ˜„์žฌ๋””๋ ‰ํ† ๋ฆฌ(.)๋กœ ์‹ค์Šต ์ฝ”๋“œ๋ฅผ ๋ณ€๊ฒฝํ•จ
  • 2025๋…„ 8์›” 25์ผ download.openmmlab.com ์‚ฌ์ดํŠธ์˜ ssl ์ด์Šˆ๋กœ pip install ์— --trusted-host ์˜ต์…˜ ๋ฐ wget์— --no-check-certificate ์˜ต์…˜ ์ถ”๊ฐ€

 

 

 

์–ดํ—ˆ.. ํŒŒ์ด์ฌ ๋ฒ„์ „์—…์œผ๋กœ ์„ค์น˜๊ฐ€ ์•ˆ๋˜์„œ ์‚ฌ์šฉ์ด ๋ถˆ๊ฐ€ํ•˜๋‹ต๋‹ˆ๋‹ค. ์ฐธ๊ณ ํ•˜๊ธฐ