NVIDIA Kaolin 库 提供用于处理各种 3D 表征的 PyTorch API,并且包含日益增多的 GPU 优化操作,例如模块化可微渲染、表征之间快速转换、数据加载、3D 检查点等。
一.下载kaolin包
https://github.com/NVIDIAGameWorks/kaolin
二.下载依赖包
个人建议新建一个虚拟环境后下载在新的虚拟环境中。
下面是主要的依赖包:
scipy<=1.5.2(注意此版本scipy要求python<=3.8) usd-core(这个包在之后下载kaolin时会自动下载) tqdm==4.61.0 flask==1.1.2 tornado==6.0.4 pytorch==1.5.0 torchvision==0.6.0 cython==0.29.20
三.下载c++解释器
因为kaolin中部分是用c++写的,所以需要这步否则会报错。
以前用过c++的可能已经下载过了可以自己找一下。
1.下载Visual Studio Code
https://code.visualstudio.com/
在刚下载的vs code中把图中蓝框中的插件下载好
2.下载Microsoft C++ 生成工具
https://visualstudio.microsoft.com/zh-hans/visual-cpp-build-tools/
将图中黄框插件勾选,建议尽量不要修改默认位置,之后下载。
四.设置kaolin
打开prompt(没有的话用cmd也行,下述均以prompt为例)
activate py36 cd C:UsersAdministratorkaolin-master python setup.py install
当出现以下语句时说明下载成功。
Finished processing dependencies for kaolin==0.10.0
此时还需要编译一下代码,输入以下语句。
python setup.py build develop
五.检验
进入python
import kaolin print(kaolin.__version__)
正确显示版本号的话说明下载完成。



