构建支持树莓派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