解决pytorch服务打包docker镜像过大问题，使用torch的cpu版本_docker镜像安装torch之后就很大

作者：你好赵伟 | 2024-06-24 11:15:48

踩

docker镜像安装torch之后就很大

使用torch的cpu版本即可，修改pip的下载依赖，在linux机器上不要默认下载

整个服务只使用了pytorch关于gcn相关的代码，但是要安装pytorch包，这是很大的。尤其如果你在linux上安装torch，会默认安装cuda版本，这会导致大小激增。我默认安装后，镜像大小8个多G，这很显然是不合理的。
后来我为了减少体积，我只安装了torch的cpu版本，即修改了requirement.txt文件

-f https://download.pytorch.org/whl/cpu/torch_stable.html
torch==2.1.0+cpu
-f https://pytorch-geometric.com/whl/torch-2.1.0+cpu.html
torch_geometric
1
2
3
4

只安装cpu版本，大小只有1.6G.

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】