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
/*进入你自己要下载kaolin的虚拟环境(py36处指你自己的虚拟环境名)。*/
cd C:\Users\Administrator\kaolin-master
/*进入kaolin文件,路径每个人不一样,只要是kaolin文件中setup.py在的目录就行。*/
python setup.py install
/*设置kaolin*/
当出现以下语句时说明下载成功。
Finished processing dependencies for kaolin==0.10.0
此时还需要编译一下代码,输入以下语句。
python setup.py build develop
五.检验
进入python
import kaolin
print(kaolin.__version__)
正确显示版本号的话说明下载完成。