| # 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 | |