Ubuntu上配置MinkowskiEngine踩坑记录

Ubuntu18.04配置MinkowskiEngine0.4.2

  • 踩坑中的相关东西
    • 相关链接
    • 检查gcc版本
    • 安装步骤
    • 小彩蛋

踩坑中的相关东西

成功配置:

Ubuntu:18.04
cuda:10.2
cudnn:7.6.5
python=3.6
pytorch=1.3.0
gcc=7.5

相关链接

cuda10.2
cudnn(选择cuDNN v7.6.5 (November 18th, 2019), for CUDA 10.2的cuDNN Library for Linux)
安装cuda10.2+cudnn7.6.5教程
MinkowskiEngine0.4.2

检查gcc版本

gcc -v //需要>7

如果gcc为5.5,一定要升级!

安装步骤

# 创建环境
conda create -n mink python=3.6
# 激活环境
conda activate mink
# 安装openblas
sudo apt install libopenblas-dev g++-7
# pytorch
conda install pytorch==1.3.0 torchvision cudatoolkit=10.1
export CXX=g++-7
cd MinkowskiEngine0.4.2
python setup.py install

小彩蛋

conda换源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
# 建议在 ~/.condarc中把-default去掉

你可能感兴趣的:(CV,ubuntu,pytorch,深度学习)