channels:
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python=3.10.*
  - pytorch==2.0.1
  - torchvision==0.15.2
  - torchaudio==2.0.2
  - pytorch-cuda=11.8
  - mkl=2024.0.0
  - pip
  - pip:
    - git+https://github.com/facebookresearch/detectron2.git
    - git+https://github.com/cocodataset/panopticapi.git
    - git+https://github.com/mcordts/cityscapesScripts.git
postinstall: |
    postinstall: |
    export CUDA_HOME=/opt/modules/nvidia-cuda-11.8.0
    export PATH=$CUDA_HOME/bin:$PATH
    export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
    module load gcc/10.5.0  # 根据集群环境调整
    export CC=gcc-10
    export CXX=g++-10
    export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=1"
    export FORCE_CUDA=1

    cd /home/yuqian_fu/Projects/PSALM
    pip install -e .
    pip install opencv-python addict numpy==1.23.5

    cd psalm/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops
    rm -rf build
    sh make.sh  # 确保make.sh包含编译扩展的命令