# conda create -n vlm python=3.10 # conda install pytorch==2.3.1 torchvision==0.18.1 pytorch-cuda=12.1 cuda -c pytorch -c "nvidia/label/cuda-12.1.0" -c "nvidia/label/cuda-12.1.1" # TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 9.0" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" CUDA_HOME=$(dirname $(dirname $(which nvcc))) LD_LIBRARY_PATH=$(dirname $(dirname $(which nvcc)))/lib MMCV_WITH_OPS=1 FORCE_CUDA=1 python -m pip install https://github.com/open-mmlab/mmcv/archive/refs/tags/v2.1.0.zip transformers==4.42.3 xtuner[deepspeed]==0.1.23 timm==1.0.9 mmdet==3.3.0 hydra-core==1.3.2 ninja==1.11.1