channels:
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python=3.10.*
  - pytorch==2.1.0
  - torchvision==0.16.0
  - torchaudio==2.1.0
  - pytorch-cuda=12.1
  - mkl=2023.*
  - llvm-openmp>=15.0.7,<16
  - pip
  - pip:
      - git+https://github.com/facebookresearch/detectron2.git
      - git+https://github.com/cocodataset/panopticapi.git
      - git+https://github.com/mcordts/cityscapesScripts.git
postinstall: |
    export CUDA_HOME=/opt/modules/nvidia-cuda-12.1.0
    export PATH=/opt/modules/nvidia-cuda-12.1.0/bin:$PATH
    export LD_LIBRARY_PATH=/opt/modules/nvidia-cuda-12.1.0/lib64:$LD_LIBRARY_PATH
    export PATH=/opt/modules/gcc-11.5.0/bin:$PATH
    export CC=gcc
    export CXX=g++
    echo "CUDA_HOME: $CUDA_HOME"
    cd /home/yuqian_fu/Projects/PSALM
    pip install -e .
    pip install opencv-python addict
    echo "GCC版本:"
    gcc --version
    pip install numpy==1.23.5
    cd psalm/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops
    python setup.py clean
    rm -rf build
    sh make.sh
    python -c "import torch; print('\n验证结果:', torch.__version__, torch.cuda.is_available(), torch.version.cuda)"