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

安装环境(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 即可安装成功

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容