专栏首页SnailTyanCaffe与NVIDIA Docker不兼容的问题

Caffe与NVIDIA Docker不兼容的问题

版权声明:博客文章都是作者辛苦整理的,转载请注明出处,谢谢! https://blog.csdn.net/Quincuntial/article/details/80833580

文章作者:Tyan 博客:noahsnail.com | CSDN | 简书

今天在使用Dockerfile构建Caffe Docker Image时碰到了一个Caffe与NVIDIA Docker不兼容的问题,我使用的NVIDIA Docker为FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04,出现的错误为:

nvcc fatal   : Unsupported gpu architecture 'compute_20'

解决方案:

# 在Dockerfile中添加
ENV CUDA_ARCH_BIN "35 52 60 61 70"
ENV CUDA_ARCH_PTX "52 70"

# 在CMake时添加参数

-DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN} -DCUDA_ARCH_PTX=${CUDA_ARCH_PTX}

注意:35 52 60 61 70是CUDA显卡的计算能力。CUDA_ARCH_BIN参数指定的是显卡的计算能力,CUDA_ARCH_PTX是PTX代码生成的对应库文件,与显卡计算能力对应。CUDA_ARCH_PTX必须包含你的显卡,否则会报错。错误如下:

Check failed: error == cudaSuccess (48 vs. 0) no kernel image is available for execution on the device

显卡计算能力查询可参考资料2。

参考资料

  1. https://github.com/NVIDIA/nvidia-docker/issues/597
  2. https://developer.nvidia.com/cuda-gpus

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 走向云原生:人们需要知道的6个基本内容

    从本质上讲,云原生是一种提高业务速度的方法,也是一种构建团队的方法,可以利用Kubernetes和容器等云原生技术提供自动化和可扩展性。

    静一
  • ASP.NET Core Jenkins Docker 实现一键化部署

    在前段时间尝试过用Jenkins来进行ASP.NET Core 程序在IIS上面的自动部署。

    DevOps时代
  • docker私有仓库搭建,证书认证,鉴权管理

    -Docker 是一个开源的应用容器引擎,让开发者可以打包他们的应用以及依赖包到一个可移植的容器中,然后发布到任何流行的 Linux 机器上,也可以实现虚拟化。

    互扯程序
  • pinpoint 修改hbase表TTL值

    缘起 Pinpoint接入业务监控后数据量大涨,平均每天Hbase数据增量20G左右,数据量太大,需要对数据进行定期清理,否则监控可用性降低,由于之前环境是由d...

    三杯水Plus
  • K8S 生态周报| 2019-04-28~2019-05-05

    > 「K8S 生态周报」内容主要包含我所接触到的 K8S 生态相关的每周值得推荐的一些信息。欢迎订阅知乎专栏「k8s生态」。

    Jintao Zhang
  • Nacos 修改mysql中默认账号密码

    官方给出一个docker镜像,就不要我们自己制作了。文档地址:https://github.com/nacos-group/nacos-docker,参考Sta...

    用户2235302
  • rancher流水线自动化部署

    准备docker环境,不会安装的参考链接 https://docs.docker.com/install/linux/docker-ce/centos/

    24-丰总
  • 『互联网架构』软件架构-企业级dubbo应用(下)(43)

    1.服务提供方启动时,向 Key:/dubbo/XXXServer/providers 下,添加当前提供者的地址。 2.并向 Channel:/dubbo/co...

    IT故事会
  • 解决因为机器性能问题导致docker-compose运行容器超时的问题

    就是一到一个docker-compose文件里面要运行的容器变多,或者因为机器性能差导致没有在60s之内把docker-compose文件中的容器完全运行起来都...

    bboysoul
  • Nacos 修改mysql中默认账号密码

    官方给出一个docker镜像,就不要我们自己制作了。文档地址:https://github.com/nacos-group/nacos-docker,参考Sta...

    码农笔录

扫码关注云+社区

领取腾讯云代金券