File size: 1,394 Bytes
36c1e62 |
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 |
# name: ego_new
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- python=3.10.*
- pytorch==2.0.1
- torchvision==0.15.2
- torchaudio==2.0.2
- gcc_linux-64=9
- gxx_linux-64=9
- 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: |
export CUDA_HOME=/opt/modules/nvidia-cuda-11.8.0
export PATH=/opt/modules/nvidia-cuda-11.8.0/bin:$PATH
export LD_LIBRARY_PATH=/opt/modules/nvidia-cuda-11.8.0/lib64:$LD_LIBRARY_PATH
export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0"
echo $CUDA_HOME
cd /home/yuqian_fu/Projects/ObjectRelator
pip install -e .
pip install opencv-python addict
echo "gcc version:"
gcc --version
cd objectrelator/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops
python setup.py clean
rm -rf build/
sh make.sh
pip install natsort
pip install scikit-image
pip install numpy==1.23.5
# variables:
# CC: /usr/bin/gcc-9
# CXX: /usr/bin/g++-9
# TORCH_NVCC_FLAGS: "--compiler-bindir=/usr/bin/gcc-9"
# export CC=/usr/bin/gcc-9
# export CXX=/usr/bin/g++-9
# export TORCH_NVCC_FLAGS="--compiler-bindir=/usr/bin/gcc-9"
# export TORCH_CUDA_ARCH_LIST="your_arch_list_here" # （可选） |