Pomprogram的博客

硬件环境 操作系统：CentOS 8.3 显卡：RTX 3090 软件环境 Pytorch版本：1.8.1 PyTorch的CUDA版本：11.1 代码/数据准备 报错处理 （1）报错1：RuntimeError: output with shape [1，256,192] doesn't match the broadcast shape [3，256，192] 解决方法：修改dataset.py 第23行代码。 修改前： 修改后： （2）报错2：RuntimeError: view size is.

如果要将我们写好的Python代码生成API接口时，我们需要借助Flask框架 1. 安装Flask pip install Flask 2. 编写Flask应用 （1）新建python文件 app.py（注意创建的Python文件名不能为flask.py，这会与Flask 本身发生冲突） （2）在新建的文件里写下如下内容： #app.py from flask import Flask app = Flask(__name__) @app.route('/test',methos=['post'])