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. 创建虚拟环境 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/ -
下载 kaolin 的代码
git clone --recursive https://github.com/NVIDIAGameWorks/kaolin # 执行此命令会同时下载 CUB # 由于服务器无法访问外网，所以我将 git 的 gitconfig 文件进行了更改，使其从国内的镜像网站下载，解决了服务器不能下载的问题 # 关于更换 git 下载源的问题，可以在另一篇博客中查看 ： https://blog.csdn.net/qq_41071191/article/details/121367039 cd kaolin -
选择需要的版本
git checkout v0.9.1 # 在这里我选择 0.9.1 的版本 -
设置环境变量用以运行 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 即可。 -
运行 setup.py
python setup.py develop运行结束可以看到此语句即说明安装成功
-
进行测试
python -c "import kaolin; print(kaolin.__version__)"可以得到对应的安装的版本。
如有问题请发私信或邮件：[email protected]