kaolin更新了0.9.0版本,请参考Kaolin Library Documentation,本博客已经完全过时了。
Kaolin是什么?
一个用于加速3D深度学习研究的PyTorch库
安装
官方给出的环境要求
1.linux
2.python==3.6
3.CUDA>=10.0.130 要装好nvcc
4.显卡驱动>410.48
5.numpy>=1.17 且 numpy<1.18.4
6. PyTorch >=1.2, <1.5 and Torchvision >=0.4.0
我的安装过程
1.环境:
Ubuntu16.04
Driver Version=418.39
CUDA Version=10.1
Gpu=2080Ti
gcc version=5.4.0
2.步骤
1.官方建议使用conda虚拟环境:
conda create -n kaolin python=3.6
conda activate kaolin
2.安装numpy,我安装的是1.17
conda install numpy=1.17
3.安装pytorch和torchvision,注意版本
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
完成后 conda list | grep numpy,看一下输出是否正常
4.下载好后解压,得到kaolin文件夹,进入kaolin
cd kaolin
5.准备工作做好,接下来是关键
python setup.py build_ext --inplace
如果这一步没有任何问题,那恭喜你
我出现的问题:
1.网络问题,一直卡在fetching python@3.6.8-linux-x86_64.tar.gz from S3 ...
这个时候就需要各显神通了,我通过ssr解决。
2.解决网络问题后,出现错误
nvcc fatal : Path to libdevice library not specified
build error:command '/usr/local/nvcc' failed with exit status 1
我的解决方法:
问题出在环境变量上,修改用户目录下面的cuda路径,
在.bashrc文件下方添加这两行,其中cuda版本对应自己电脑装的版本
export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
再次python setup.py build_ext --inplace,成功
6.上述问题成功解决的话,基本已经成功,最后再输入
python setup.py install
我没有碰到问题,就等他装好
7,装好后测试一下
python
import kaolin as kal
print(kal.__version__)
如果出现问题:ModuleNotFoundError: No module named 'nuscenes'
则:pip install --no-dependencies nuscenes-devkit opencv-python-headless scikit-learn joblib pyquaternion cachetools
转载请注明原文地址:https://blackberry.8miu.com/read-30641.html