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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

shan_shmily

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值