为了用GPU加速TENSORFLOW,笔者折腾了两天,终于给我的双显卡笔记本安装上了CUDA,期间电脑几次进不了桌面,说实话,我的内心是崩溃的。
尝试了.run文件和直接apt-get安装,最后使用的方法是apt。在此将过程写下来,免得小白入坑。
笔者的环境:
操作系统:
Ubuntu 16.04 LTS desktop amd64
显卡和CPU:
Intel i7-7500 Nvidia-Geforce-940MX
安装过程:
一:检查你的显卡是否支持CUDA,参考网址
http://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#verify-you-have-cuda-enabled-system
二:这个教程采用apt网络安装方式
- 升级系统
apt-get update && apt-get upgrade
- 下载并安装VitualGL (https://sourceforge.net/projects/virtualgl/files/)
dpkg -i virtualgl*.deb
下载并安装CUDA8.0 ( https://developer.nvidia.com/cuda-downloads )
建议通过如下方式
安装依赖.
apt-get install linux-headers-$(uname -r)
apt-get install freeglut3-dev libxmu-dev libpcap-dev
- 编辑 .bashrc 加入PATH。下面变量的设置可能要根据你的环境而定
export PATH=$PATH:/opt/VirtualGL/bin
export PATH=$PATH:/usr/local/cuda/bin
- 安装bumblebee-nvidia 和 primus.它支持在Intel和Nvidia显卡间自动分配功率
apt-get install bumblebee