官网:https://tensorflow.google.cn/
安装 NVIDIA 驱动
卸载现有的 nvidia 所有驱动
sudo apt-get remove--purge nvidia*
禁用 nouveau:
sudo gedit /etc/modprobe.d/blacklist.conf 在最后一行添加:blacklist nouveau
sudo update-initramfs-u
重启电脑之后
lsmod|grep nouveau #没有输出,即说明安装成功
安装驱动:
Ctrl+Alt+F1,之后输入用户名和密码登录即可。
cd Software
sudo service lightdm stop
sudo chmod a+x NVIDIA-Linux-x86_64-384.111.run(可以按 Tab 键补全)
sudo ./NVIDIA-Linux-x86_64-384.111.run -no-opengl-files #不安装 OpenGL
sudo service lightdm start
nvidia-smi 可以看显卡情况
cuda 安装(安装 cuda9.0)
卸载原先安装的 8.0
cd /usr/local/cuda/bin sudo ./uninstall_cuda_8.0.pl
安装
下载的目录打开终端,输入如下指令安装。sudo sh cuda_9.0.176_384.81_linux.run
按照终端的提示一步步操作即可:
按 q 退出条款浏览或者按空格直到条款末尾,输入 accept 接受条款。
输入 n 不安装 nvidia 图像驱动,因为前面已经安装好驱动。如果这里按照默认选择 yes,则
前面安装的高版本 nvidia 驱动将被替换成为 cuda 里带的低版本 nvidia 驱动。
输入 y 安装 cuda 9.0 toolkit。
回车确认 cuda 默认安装路径(/usr/local/cuda-9.0)。
输入 y 安装 CUDA 9.0 Samples。
输入 CUDA 9.0 Samples 安装路径:/home/用户名/CUDA/samples。
至此 CUDA 安装完成,下面需要配置 CUDA 的环境变量,保证其正常工作。
打开新终端,设置 CUDA 环境变量:
sudo gedit /etc/profile
如果在刚才的安装过程中均默认安装路径,则把下面内容复制到打开的文件中:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH