-
摘要:
最近在学习搭建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