이거 다시 깔기
인자 맞춰주기
mamba_ssm/ops/selective_scan_interface.py
만약 mamba도 param으로 인식되면 걍 하면 됨 → 얘도 파이토치 기반이기 때문에 파라미터가 인식된다고 한다.
아니라면 폴더 만들어서 따로 처리하는 게 편할듯
/usr/local$ ls에 CUDA 있다.
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run sudo sh cuda_11.7.0_515.43.04_linux.run
가상환경은 python 3.9로
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
cd detectron2 pip install -e .
git clone https://github.com/state-spaces/mamba.git cd mamba pip install --no-build-isolation mamba-ssm==1.0.0
pip install causal-conv1d==1.4.0
conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(x, conv1d_weight, conv1d_bias, None, None, None, True)