记录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版本,另一条路就是换模型。。。。。