zhusuan环境搭建【傻瓜式教程】

概要

zhusuan是清华大学朱军副教授课题组开发的贝叶斯深度学习的GPU库，本文章是详细介绍了如何在自己的Linux系统上搭建该工具的步骤。

整体搭建流程

1. 下载zhusuan工程

// stable版本
git clone https://github.com/thu-ml/zhusuan.git

2. 搭建虚拟环境，并安装指定版本的Tensorflow

由于zhusuan工程是基于Tensorflow搭建的，且要求版本为1.13.0以上，搭建步骤为

conda create -n tf python=3.7       # 创建虚拟环境tf，并安装python3.7          

创建完成后，激活虚拟环境，并安装tensorflow

conda activate tf      # 激活环境
conda install tensorflow=1.15.0      #conda直接安装，不成功考虑镜像源的问题      

2. 安装库依赖

在虚拟环境 tf 中，索引到工程目录，并安装相应的依赖

cd zhusuan
pip install .
pip install ".[examples]"         

完成依赖安装之后，即可开始跑例程，例程在

cd examples/bayesian_neural_nets
python bnn_vi.py    

值得注意的是！！！
这里会存在报错的问题，显示不存在examples库，解决方式为，在bnn_vi.py中添加代码：

import zhusuan as zs

import sys
sys.path.append('/home/cq/chen_BNN/zhusuan')       #更换为自己的zhusuan目录即可

from examples import conf   

实现效果

如图即为成功跑通例程：