简单实现Ubuntu16.04 + caffe2 + CUDA9.0 + cuDNN8.0

在Ubuntu16.04 CUDA9.0 cuDNN8.0的环境下安装caffe2

本博客比较简单,cuda9.0 cudnn8.0部分请看上一篇博客,其中详细讲了:

  • 如何安装驱动
  • 安装cuda
  • 安装cudnn
  • 安装tensorflow

本教程主要参考来自Caffe2官方说明文档

For GPU support you will need CUDA, CuDNN, and NCCL. These must be installed from Nvidia’s website.

选择ubuntu,Pre-built Binaries,其中有一个安装方式:

For Caffe2 with CUDA 9 and CuDNN 7 support:

conda install -c caffe2 caffe2-cuda9.0-cudnn7

但是我们要先将NCCL安装,选择NCCL 2.2.13 for Ubuntu 16.04 and CUDA 9,下载下来的文件是deb格式的

我的全称为:nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb

开始安装:

sudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb

最后在你的conda环境下运行conda install -c caffe2 caffe2-cuda9.0-cudnn7即可

posted @   pprp  阅读(1232)  评论(0编辑  收藏  举报
编辑推荐:
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· .NET 9 new features-C#13新的锁类型和语义
阅读排行:
· Sdcb Chats 技术博客:数据库 ID 选型的曲折之路 - 从 Guid 到自增 ID,再到
· 语音处理 开源项目 EchoSharp
· 《HelloGitHub》第 106 期
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 使用 Dify + LLM 构建精确任务处理应用
历史上的今天:
2017-08-27 codeforces 578c - weekness and poorness - 三分
2017-08-27 最大连续区间和问题
2017-08-27 二维前缀和 - 算法学习 - 输入输出优化
2017-08-27 ACM输入函数测试 - scanf cin 优化的输入
2017-08-27 一维前缀和 - 包含输入输出优化
点击右上角即可分享
微信分享提示