代码复现:KPConv

本文介绍了复现KPConv代码过程中遇到的环境配置问题,包括TensorFlow 1.12.0和CUDA 9.0的版本要求,以及在复现过程中遇到的依赖冲突、编译错误和显存不足等具体问题的解决方案。针对显存不足,通过降低训练批大小解决了在2080ti显卡上运行的问题。
摘要由CSDN通过智能技术生成
说一下我复现KPConv代码的环境和遇到的一些问题

1、tensorflow 1.12.0 和 CUDA 9.0。这是官方指导意见。如果代码运行时遇到各种各样的如: 'tensorflow' has no attribute 'xxx'、找不到tf.math.xxx等问题,基本就是tensorflow和cuda版本问题,或者是版本冲突。另外,需要重装tensorflow的时候,记得完全卸载它,用pip或conda安装的要用对应的卸载方式,还要注意cpu和gpu版本,还要卸载protobuf。卸载完成之后,记得用conda list查看一下有没有卸载干净,然后再装新的版本,不然会遇到各种各样的问题。

2、运行时遇到这样的错误

tensorflow.python.framework.errors_impl.NotFoundError: tf_custom_ops/tf_neighbors.so: un
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值