如何搭建DCVC-RT深度条件视频压缩环境?

摘要:安装环境(install env) git clone https:github.commicrosoftDCVC.git dcvc-rt cd dcvc-rt conda create -n dcvcrt python=3.12
安装环境(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 构建工程(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)如果执行pip install . 有报错 ModuleNotFoundError: No module named 'pybind11'. 但是安装 pybind11 后扔然报错,可以通过如下方式进行解决: 在src/cpp/目录下新建一个pyproject.toml文件 vim pyproject.toml 文件内写入如下内容: [build-system] requires = ["setuptools", "wheel", "pybind11"] build-backend = "setuptools.build_meta" 然后再执行pip install . 即可 (2)如果执行 pip install . --no-build-isolation 有报错: 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 # 创建指向12.6的新软链接 nvcc -V # 查看软链接是否更换成功 更换成功之后,再执行 pip install . --no-build-isolation 即可安装成功