Ubuntu18.04 安装 kaolin 库

Kaolin 是一套用于加速 3D 深度学习研究的工具。具体介绍可在官方网站查阅，本博客不加以赘述。官方网站为：https://kaolin.readthedocs.io/en/latest/

一、Ubuntu系统

当前的 Ubuntu 操作系统版本为：18.04
已知当前服务器有两张显卡，分别为： Quadro RTX 6000 、 Quadro P5000
请确保已正确安装 Nvidia 显卡驱动 、NVCC 、re2c

二、C / C++ 环境准备

因为 kaolin 部分代码使用 C++ 语言，因此，我个人认为 C/C++ 的编译环境是必要的。

1. 更新 apt
sudo apt update
2. 安装 gcc g++ make
sudo apt install gcc g++ make
3. 安装依赖库
sudo apt install libglu1-mesa libxi-dev libxmu-dev libglu1-mesa-dev freeglut3-dev

三、kaolin 环境准备

1. 通过以下命令安装依赖包

   1. 创建虚拟环境
   conda create -n kaolin python=3.7
   conda activate kaolin
   
   2. 安装以下依赖包
   conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
   # 该命令可以在 pytorch 官方网站找到，网站地址为:  https://pytorch.org/get-started/previous-versions/
   # 可以在网站中选择需要的版本的安装命令
   
   pip install Cython==0.29.20 -i https://pypi.douban.com/simple/
   # 官网指令，临时更改为豆瓣源，此包不能更改为其他版本，在 setup 文件中会对此包的版本进行限制
   # Cython 的主要作用为编译 python 语言为 C 语言
   
   pip install scipy -i https://pypi.douban.com/simple/
   # 下载会得到 1.7.1 的版本，符合环境要求
   
   pip install Pillow -i https://pypi.douban.com/simple/
   # 下载会得到 8.4.0 的版本，符合环境要求
   
   pip install usd-core==20.11 -i https://pypi.douban.com/simple/
   

2. 下载 kaolin 的代码

   git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
   # 执行此命令会同时下载 CUB
   # 由于服务器无法访问外网，所以我将 git 的 gitconfig 文件进行了更改，使其从国内的镜像网站下载，解决了服务器不能下载的问题
   # 关于更换 git 下载源的问题，可以在另一篇博客中查看 ：
   https://blog.csdn.net/qq_41071191/article/details/121367039
   
   cd kaolin
   

3. 选择需要的版本

   git checkout v0.9.1
   # 在这里我选择 0.9.1 的版本
   

4. 设置环境变量用以运行 setup.py

   export KAOLIN_INSTALL_EXPERIMENTAL=1
   
   export USE_CYTHON=1
   # 确保 Cython 已经正确安装并且设置环境变量
   
   export TORCH_CUDA_ARCH_LIST="6.1 7.0"
   # 如果使用的是异构的 GPU 需要根据实际架构设置环境变量，经查，P5000 的算力为 6.1 ，RTX 6000 的算力为 7.5 。但设置为 6.1 和 7.5 仍会报错，查阅资料后，将 7.5 改为 7.0 即可。
   

5. 运行 setup.py

   python setup.py develop
   

   运行结束可以看到此语句即说明安装成功
   

6. 进行测试

   python -c "import kaolin; print(kaolin.__version__)"
   

   可以得到对应的安装的版本。

如有问题请发私信或邮件：xukuo11@foxmail.com