在Ubuntu上安装CUDA

首先

我想尝试一下DeepLearning,所以决定安装CUDA。虽然有很多前辈的经验,能找到很多在Ubuntu上安装CUDA的文章,但按照文章的步骤却没有成功安装。为了方便那些也在努力安装的人,我总结了自己成功安装的方法,希望能对他人有所帮助。

省略使用的語言,仅保留目标

在Ubuntu 16.04LTS(搭载GeForce GTX 750)上安装CUDA 8。

下载Ubuntu操作系统
下载CUDA工具包

CUDA Toolkit 安装步骤

假设Ubuntu已经安装完成。

1. 准备事项

因为nouveau启用时无法正常安装,所以需要禁用nouveau。
根据CUDA Toolkit安装手册,引用了禁用nouveau的方法。

    请创建文件/etc/modprobe.d/blacklist-nouveau.conf,并写入以下字符串

/etc/modprobe.d/blacklist-nouveau.conf
黑名单 nouveau
选项方式 nouveau modeset=0

重新构建Initramfs。

请在终端输入以下命令进行更新:
sudo update-initramfs -u

3. 重新启动

$ sudo reboot

4. BIOS設定でUEFI無効またはセキュアブートを無効にする

2. CUDA Toolkit インストール

Download CUDA Toolkit から、 runfile(local) をダウンロード
2. ディスプレイマネージャが起動しているとインストールできないため、停止させる

$ sudo service lightdm stop

3. Ctrl+Alt+F1 でCUIモードログインをし、1. でダウンロードした runfile(local) を実行してインストール
4. Ubuntu 16.04へのCUDAインストール方法 より、パスの設定を行う。設定ファイルと設定方法を以下に引用する。

~/.bashrc
将”/usr/local/cuda-8.0/bin”添加到路径中:export PATH=”/usr/local/cuda-8.0/bin:$PATH”
将”/usr/local/cuda-8.0/lib64″添加到库路径中:export LD_LIBRARY_PATH=”/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH”

5. 重新启动

$ sudo reboot

若能通过GUI登录,则表示安装已完成。如果无法完成登录,则继续进行下一步。

3. 重新安装NVidia显卡驱动程序

执行以下操作以在Ubuntu上安装Nvidia GPU的驱动程序 + CUDA(适用于GTX1080):2. 3. 4. 5. 6. 引用并编辑结果如下
1. 删除旧驱动程序

使用sudo apt-get –purge remove nvidia-*删除NVIDIA驱动程序。

2. 注册存储库

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update

请运行以下命令添加apt仓库:
$ sudo add-apt-repository ppa:graphics-drivers/ppa
然后更新apt:
$ sudo apt-get update

3. 安装适用于GPU的驱动程序

请使用以下命令安装Nvidia-375:sudo apt-get install nvidia-375

驱动程序的版本将是在通过“Download Drivers”搜索相应的GPU版本时找到的版本。

4. 重新启动

请使用超级用户模式重新启动。

确定

$nvidia-smi: 英伟达系统管理接口

如果没有特别错误的话,就是可以的。

广告
将在 10 秒后关闭
bannerAds