如何在Databricks中安装Graphviz？pip安装失败求助

一、安装系统级Graphviz依赖 #

pip仅能安装Graphviz的Python绑定库，Databricks集群还需要底层的系统级Graphviz包，可通过两种方式安装：

• 临时会话安装（仅当前Notebook会话有效）：
  在Notebook单元格中执行Shell命令：

%sh
apt-get update && apt-get install -y graphviz

• 永久集群安装（所有会话生效）：

1. 创建Shell脚本install_graphviz.sh，内容为上述apt-get命令；
2. 将脚本上传至DBFS（如dbfs:/scripts/install_graphviz.sh）；
3. 编辑集群配置，在"初始化脚本"中添加该脚本路径，重启集群后生效。

二、安装Python Graphviz绑定包 #

系统依赖安装完成后，执行pip命令安装Python库：

%pip install graphviz

三、解决Databricks图表显示问题 #

直接调用Graphviz的默认可视化方法无法在Databricks中显示，可通过以下两种方式输出：

• 方式1：导出为图片文件后显示

from graphviz import Digraph

# 创建示例图
dot = Digraph()
dot.node('A', 'Hello')
dot.node('B', 'World')
dot.edge('A', 'B')

# 导出为PNG并清理临时文件
dot.render('graph-output.gv', format='png', cleanup=True)

# 在Databricks中显示图片
display(dbutils.fs.ls('/databricks/driver/graph-output.gv.png'))

• 方式2：生成SVG直接显示

from IPython.display import display, SVG
from graphviz import Digraph

dot = Digraph()
dot.node('A', 'Hello')
dot.node('B', 'World')
dot.edge('A', 'B')

# 生成SVG内容并显示
svg_data = dot.pipe(format='svg').decode('utf-8')
display(SVG(svg_data))

内容的提问来源于stack exchange，提问作者Fred_Xu