安裝環(huán)境(install env)
git clone https://github.com/microsoft/DCVC.git dcvc-rt
cd dcvc-rt/
conda create -n dcvcrt python=3.12
conda activate dcvcrt
pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126
pip install -r requirements.txt
構(gòu)建工程(build the project)
sudo apt-get install cmake g++ ninja-build
cd ./src/cpp/
pip install .
cd ../layers/extensions/inference/
pip install . --no-build-isolation
(1)如果執(zhí)行pip install . 有報(bào)錯(cuò)
ModuleNotFoundError: No module named 'pybind11'.
但是安裝 pybind11 后扔然報(bào)錯(cuò),可以通過如下方式進(jìn)行解決:
在src/cpp/目錄下新建一個(gè)pyproject.toml文件
vim pyproject.toml
文件內(nèi)寫入如下內(nèi)容:
[build-system]
requires = ["setuptools", "wheel", "pybind11"]
build-backend = "setuptools.build_meta"
然后再執(zhí)行pip install . 即可
(2)如果執(zhí)行 pip install . --no-build-isolation 有報(bào)錯(cuò):
RuntimeError:
The detected CUDA version (11.8) mismatches the version that was used to compile
PyTorch (12.6). Please make sure to use the same CUDA versions.
原因是 nvcc -V 的版本不匹配
使用如下命令將軟鏈接更換為 12.6
sudo rm -f /usr/local/cuda # 刪除舊的軟鏈接
sudo ln -s /usr/local/cuda-12.6 /usr/local/cuda # 創(chuàng)建指向12.6的新軟鏈接
nvcc -V # 查看軟鏈接是否更換成功
更換成功之后,再執(zhí)行 pip install . --no-build-isolation 即可安裝成功