Caffe简明教程2：安装Caffe的第一步-安装CUDA

Posted 2023-03-04 xietx1995

您可以查看所有文章的索引：Caffe简明教程0：文章列表

---

本文介绍如何在 Ubuntu16.04 上安装 CUDA8.0。
几点说明：

• 如果需要GPU加速的话，安装Caffe前需要安装CUDA8.0以及cuDNN5.1。
• 我使用的操作系统是Ubuntu 16.04 LTS，如果安装过程中遇到任何问题，请加群（博客左上角）讨论。
• 如果你想在其他操作系统上安装Caffe，请参考官方文档。

---

1.1 下载CUDA

首先在此处下载 CUDA8.0，打开页面后，选择操作系统对应的版本下载，这里我们需要ubuntu 16.04版本：

选择版本后，下载下面两个文件，第一个是CUDA的 .deb安装包，另一个是cuBLAS的补丁：

下载完成后，在我的电脑上得到了这两个文件：

---

1.2 安装CUDA8.0

1）NVIDIA显卡驱动

安装CUDA前需要安装NVIDA显卡驱动，如果没有安装，可以输入如下命令安装：

• sudo apt-get install nvidia-384

完成后输入命令reboot重启计算机即可。

2）安装CUDA8.0

在你下好的cuda安装包目录下，输入如下命令进行安装：

• sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
• sudo apt-get update
• sudo apt-get install cuda

如果未报任何错误即安装成功。

---

1.3 安装CUDA8.0对应的cuBLAS补丁

CUDA主体安装完成后，输入如下命令安装cuBLAS的补丁：

• sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb

如果没报错，那么CUDA就安装成功了。

---

下一篇文章：Caffe简明教程3：安装Caffe的第二步-安装cuDNN

---

欢迎加群交流，点击链接加入群【Python爱好者交流】