DCVC-RT 深度條件視頻壓縮環(huán)境搭建

安裝環(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 即可安裝成功

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容