YaoFANGUK / video-subtitle-remover

基于AI的图片/视频硬字幕去除、文本水印去除,无损分辨率生成去字幕、去水印后的图片/视频文件。无需申请第三方API,本地实现。AI-based tool for removing hard-coded subtitles and text-like watermarks from videos or Pictures.
Apache License 2.0
4.68k stars 612 forks source link

torch 版本冲突 #65

Open nbu2024 opened 6 months ago

nbu2024 commented 6 months ago

conda install pytorch==2.0.1 torchvision==0.16.0 pytorch-cuda=11.8 -c pytorch -c nvidia 用这个安装后,在安装requirements.txt,如图所示,会安装torch-2.3.0 , 虽然还没有安装完,但能确定如果使用torch会造成版本冲突,有两个版本的torch在conda环境里。

image
idrisrupt commented 5 months ago

i figured it out by doing this :

// Try with cuda 11.7 conda install paddlepaddle-gpu==2.6.0 cudatoolkit=11.7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge conda install pytorch-cuda=11.7 -c pytorch -c nvidia pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu117