DeepPlaque/environment.yml at main · ChengJin-git/DeepPlaque · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# DeepPlaque Conda Environment
#
# Create environment:
#   conda env create -f environment.yml
#
# Activate:
#   conda activate deepplaque

name: deepplaque
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults

dependencies:
  # Python
  - python=3.11

  # PyTorch with CUDA 12.1
  - pytorch=2.4.0
  - torchvision
  - pytorch-cuda=12.1

  # System libraries
  - libgl
  - glib

  # pip packages
  - pip
  - pip:
      # mmcv must be installed from OpenMMLab wheel for CUDA support
      # Choose the correct wheel for your CUDA/PyTorch version:
      # https://download.openmmlab.com/mmcv/dist/index.html
      - mmcv==2.2.0 --find-links https://download.openmmlab.com/mmcv/dist/cu121/torch2.4/index.html

      # OpenMMLab components
      - mmengine==0.10.5
      - mmdet==3.3.0

      # Image processing
      - opencv-python>=4.10.0
      - Pillow>=10.0.0
      - pycocotools>=2.0.0

      # Data processing
      - numpy==1.26.4

      # Visualization
      - matplotlib>=3.7.0

      # Model download from HuggingFace
      - huggingface_hub>=0.20.0