【搭建机器学习环境】安装Cuda 9.2

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/sks/article/details/81126380
  • 摘要:

最近在学习搭建TensorFlow,用到GPU运算,所以需要安装Cuda。

硬件环境:i7 5820K;GTX 980 *2;32GB;250GSSD

系统:Ubuntu16.04.4 64位 Desktop版

  • 参考文章:

http://docs.nvidia.com/cuda/cuda-installation-guide-linux/

  • 下载页面:

https://developer.nvidia.com/cuda-downloads

Operating System:Ubuntu

Version:16.04

Installer Type:deb (local)

  • 安装步骤:

1.Install repository meta-data

sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb

2.Installing the CUDA public GPG key

在上个步骤结尾处有提示,给出了下面的命令

sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub

3.执行安装

sudo apt-get update

sudo apt-get install cuda

4.设置环境

将环境变量填到.bashrc中
export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

5.确认驱动程序

cat /proc/driver/nvidia/version

展开阅读全文

没有更多推荐了,返回首页