- 博客(2)
- 收藏
- 关注
原创 配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
解决方法:将ssd_pascal.py文件中gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。因为电脑只安装了一块GPU。还需要将ssd_pascal.py文件中中的batch_size = 32和accum_batch_size = 32同时改为8,不然会出现“Check failed: error == cudaSuccess (2 vs. 0)...
2019-01-10 17:01:01 779
原创 Ubuntu16.04 编译 caffe 出现错误:Unsupported gpu architecture 'compute_20'
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D CUDA_GENERATION=Kepler ..
2019-01-07 08:31:35 817 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人