PointContrast和MinkowskiEngine环境cuda10.1(未完)

记录pointcontrast的心里路程,从配环境开始遇到的各种问题。

注意事项

配环境

PointContrast项目作者不维护了,然后作者后续出了升级版的,github地址ContrastiveSceneContexts
第一步配环境要求安装MinkowskiEngine0.4.3版本,源项目的cuda是10.2,由于后续安装需要使用setup.py,这里一开始我在环境里面安装的也是10.2版本,但是后面遇到的各种错误表明,setup.py安装的时候走的cuda是系统的cuda版本(使用命令nvcc --version可查看系统的cuda版本),而平时用的nvidia-smi命令输出的cuda版本是支持最高的cuda版本,所以又去重新配环境,下载和系统一样版本的cudatoolkit10.1,然后本地下载MinkowskiEngine,进行配置,依然报错:nvcc fatal:unsupported gpu architecture ‘computer_30’
在这里插入图片描述
然后网上大部分意思是编译不支持compute_30,找到Makefile文件然后注释掉这一行,然后报下面的错误
在这里插入图片描述
目前网上大部分安装成功的cuda版本都是10.2,由于用的是学校服务器,系统的cuda是不能升级了,。

本次使用的版本

  • ubuntu20.04
    (Ubuntu20.04自带的gcc版本为9.3,而cuda10.1不支持gcc-9,因此要手动安装gcc-7,找师兄安装,自己没权限)
  • gcc7.4
  • cuda10.1(CUDA 10.2, must use GCC < 8)
  • torch 1.5.1
  • MinkowskiEngine0.4.3不成功,0.5.4最新版本的可以成功
    (但是最新版本的项目有一些错误,或许可以从这条路走)

一些命令

linux查找文件和特定内容的文件

find | grep  luoluo

将当前目录下(包括子目录)的文件名中含有luoluo的文件过滤出来
即使是当前目录子目录下的文件,也能被找出来。

find | xargs grep compute_30

将当前目录下(包括子目录)的文件内容中含有compute_30的行过滤出来
查看pytorchgpu是否可用

python
import torch
print(torch.cuda.is_available())#输出True

环境变量

export CXX=g++-7
export CUDA_HOME=/usr/local/cuda-10.1
export CUDA_HOME=/usr/local/cuda
export OMP_NUM_THREADS=12
(source ~/.bashrc 激活命令)

测试MinkowskiEngine安装

#本地安装下载压缩包,各种版本的Minkowski压缩包地址https://github.com/NVIDIA/MinkowskiEngine/releases
wget https://github.com/NVIDIA/MinkowskiEngine/archive/refs/tags/v0.4.3.zip
cd MinkowskiEngine-0.4.3
python setup.py install
import MinkowskiEngine as ME
print(ME.__version__)#0.5.4

安装pytorch1.5版本的

如果需要降版本这是1.5的
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

现在还有两条路,修改项目代码使得0.5.4版本的适用0.4.3版本,另一条路就是换模型。。。。。

猜你喜欢

转载自blog.csdn.net/shan_5233/article/details/128080766