Closed JasonZhu-WHU closed 3 years ago
在本机Docker的container上可以正常运行并输出结果
有一个要注意的地方就是dockerfile第15行这里的安装pytorch代码： RUN pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html 用的是cpu的pytorch，具体部署的时候可能这里需要修改一下，来适配gpu相应的版本（gpu的环境我没试过）
RUN pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
另外改了一个返回结果代码的小bug，摘要返回目录应该是output/xxxx/xxxx.mp4
用Docker把模型环境打包，具体可以参考dockerfile和requirements
在本机Docker的container上可以正常运行并输出结果
有一个要注意的地方就是dockerfile第15行这里的安装pytorch代码：
RUN pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html用的是cpu的pytorch，具体部署的时候可能这里需要修改一下，来适配gpu相应的版本（gpu的环境我没试过）另外改了一个返回结果代码的小bug，摘要返回目录应该是output/xxxx/xxxx.mp4