Ubuntu16.04 安装Kaolin深度学习

    科技2024-04-18  108

    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
    Processed: 0.023, SQL: 9