学了自然要用,其实docker除了常见的devops套路,还可以跑AI框架之类的,免去了很多配置上的烦恼(当然,我准备做loT结合AI,对这类虚拟化平台简直是大爱啊!)
PaddlePaddle准备发车
官方懒得要死,官网上没说清楚docker怎么玩,只能自己动手了。
先弄到PaddlePaddle的镜像
Docker pull paddlepaddle/paddle:latest
安装完后,就把官网的案例下载到本地。然后新建一个文件夹workspace,在workspace下新建文件
housing.py
然后打开这个文件,把官网给的代码给复制进去
import paddle.v2 as paddle
# Initialize PaddlePaddle.
paddle.init(use_gpu=False, trainer_count=1)
# Configure the neural network.
x = paddle.layer.data(name='x', type=paddle.data_type.dense_vector(13))
y_predict = paddle.layer.fc(input=x, size=1, act=paddle.activation.Linear())
with open('/workspace/fit_a_line.tar', 'r') as f:
parameters = paddle.parameters.Parameters.from_tar(f)
# Infer using provided test data.
probs = paddle.infer(