File size: 1,164 Bytes
625a17f |
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 |
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)"
|