我正在尝试使用Flask + PyTorch运行一个基本应用程序,并将其托管在Heroku上。但是,我遇到的问题是,免费版本的最大子弹大小为500mb,而PyTorch本身约为500mb。
在Google进行一些搜索之后,有人写了一篇关于找到仅CPU版本的PyTorch的文章,并使用它,在这里要小得多。。
但是,我对如何完成此操作非常迷惑,而该人员根本没有记录在案。任何建议表示赞赏,谢谢。
编辑:
为了更具体地说明我的问题,据我所知,我尝试安装torch by,其中包括require.txt,其中将torch列为依赖项。当前我有:火炬== 0.4.1。但是,这不起作用的大小。
我的问题是,您是否知道我可以在需求文件中写些什么来获得较小的仅用于cpu的割炬版本,或者,如果requirements.txt不适用于该版本,我会怎么做,以便获取CPU版本。