构建支持树莓派3B+的Pytorch镜像

构建支持树莓派3B+的Pytorch镜像。(Python3.7,Pytorch1.7)

dockerfile文件内容如下:

FROM python:3.7.9-slim-buster

COPY torch-1.7.0a0-cp37-cp37m-linux_aarch64.whl /tmp
RUN pip3 install /tmp/torch-1.7.0a0-cp37-cp37m-linux_aarch64.whl
RUN pip3 install -i https://mirrors.aliyun.com/pypi/simple torchvision

CMD ["/bin/bash"]


构建镜像后需要安装依赖库libopenmpi-dev:

apt-get update 
apt-get upgrade
apt-get install libopenmpi-dev


apt-get update 
apt-get upgrade (期间需要输入"y"并回车)
apt-get install libopenmpi-dev(期间需要输入"y"并回车)
commit生成新镜像。

docker commit -a="hzw" -m="enable for pytorch" 3d04c4a875e0 hzw/hzwtorch:3.0

为了简便可以在构建dockerfile时直接将上述三个命令写入dockerfile中,只需要在命令apt-get 中加入-y即可,完整版的dockerfile文件内容如下:

FROM python:3.7.9-slim-buster

COPY torch-1.7.0a0-cp37-cp37m-linux_aarch64.whl /tmp
RUN pip3 install /tmp/torch-1.7.0a0-cp37-cp37m-linux_aarch64.whl
RUN pip3 install -i https://mirrors.aliyun.com/pypi/simple torchvision
RUN apt-get update
RUN apt-get -y upgrade 
RUN apt-get install -y libopenmpi-dev 

CMD ["/bin/bash"]

注意,需要将torch.whl文件事先拷贝到dockerfile的相同目录下,然后使用COPY保留字将其复制到base镜像的tmp文件夹当中。

最后保存dockerfile文件,使用如下命令build,生成镜像,需要等待一段时间。

docker build -f /home/ubuntu/mydocker/dockerfile3 -t hzwtorchpro:1.0 .

torch-1.7.0a0-cp37-cp37m-linux_aarch64.whl链接:https://pan.baidu.com/s/1BVek_zioYn31Z27Uii1Ckw 
提取码:6wy7 

已经构建好的阿里云镜像可以直接pull:

sudo docker pull registry.cn-hangzhou.aliyuncs.com/zwhuang/hzwtorch:4.0


 

 

 

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值