caffe GPU版本编译遇到的几个问题

本文详细记录了在Ubuntu18环境下编译Caffe-GPU版本时遇到的问题及解决方案,包括处理不支持的GPU架构、make runtest错误、pycaffe导入失败以及创建LMDB数据库时的文件打开失败等问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

caffe GPU版本编译遇到的几个问题

1.环境:ubuntu18 desktop版
Caffe-GPU编译问题:nvcc fatal : Unsupported gpu architecture ‘compute_20’:
https://blog.youkuaiyun.com/kemgine/article/details/78781377

2.make runtest时出现错误:

[FAILED] BatchReindexLayerTest/3.TestGradient, where TypeParam = caffe::GPUDevice<double>

解决方法:
https://blog.youkuaiyun.com/m0_38082419/article/details/79379066

3.编译pycaffe成功但是import caffe出错:

  from google.protobuf.internal import enum_type_wrapper

原因:依赖项没装好,此处为protobuf没装好,所以:
sudo install protobuf
并加入路径:

export PYTHONPATH=/work/project/caffe/python:$PYTHONPATH

4.Creating lmdb…
F0225 16:39:23.937786 23330 convert_mnist_data.cpp:45] Check failed: image_file Unable to open file data/mnist/train-images-idx3-ubyte
*** Check failure stack trace: ***
@ 0x7f5d0f0e8daa (unknown)
@ 0x7f5d0f0e8ce4 (unknown)
@ 0x7f5d0f0e86e6 (unknown)
@ 0x7f5d0f0eb687 (unknown)
@ 0x4045df convert_dataset()
@ 0x402943 main
@ 0x7f5d0de99ec5 (unknown)
@ 0x40298c (unknown)
@ (nil) (unknown)
Aborted (core dumped)
F0225 16:39:24.025714 23334 convert_mnist_data.cpp:45] Check failed: image_file Unable to open file data/mnist/t10k-images-idx3-ubyte
*** Check failure stack trace: ***
@ 0x7faa1af57daa (unknown)
@ 0x7faa1af57ce4 (unknown)
@ 0x7faa1af576e6 (unknown)
@ 0x7faa1af5a687 (unknown)
@ 0x4045df convert_dataset()
@ 0x402943 main
@ 0x7faa19d08ec5 (unknown)
@ 0x40298c (unknown)
@ (nil) (unknown)
Aborted (core dumped)
Done.
`

rename your mnist dataset,because you downloaded mnist dataset maybe was “t10k-images.idx3-ubyte” but it is the “t10k-images-idx3-ubyte” in create_mnist.sh,so you have to use '-'instead of ‘.’

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值