环境配置:Ubuntu+cuda+cudnn
驱动与安装前准备
(以下步骤也可参考官方适用于Linux的NVIDIA CUDA安装指南)
1. 验证自己的电脑是否有一个可以支持CUDA的GPU
lspci | grep -i nvidia
2. 安装NVIDIA驱动
这里可以先跳过,在之后安装cuda9.2的时候里面可以一起安装。
如果你是别的版本或者想自己尝试别的方法也可以试试下面的设置:
可以自己本地设置:
右上角>系统设置>软件和更新>附加驱动>选择NVIDIA那项>应用更改
或者下载官方的驱动自己安装
3. 验证自己的Linux版本是否支持 CUDA(Ubuntu 16.04没问题)
uname -m && cat /etc/*release
4. 验证系统是否安装了gcc
(注意所安装的gcc版本要满足官方CUDA所需gcc版本)
根据要求,我这里安装的是gcc5.4.0的版本。
gcc -v