踩坑中的相关东西
成功配置:
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去掉