在安装cuda之前,最重要的就是检查你的显卡是不是支持cuda安装版本。官网给出说明是只要在cuda支持显卡列表里的显卡就可以,但是我的显卡NVS 310虽然在,安装cuda以后会报错,因为显卡不支持。所以我猜测这个显卡或许可以支持低版本的cuda。
官网的安装文档:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/
显卡列表:https://developer.nvidia.com/cuda-gpus
官方文档在安装之前给了很多预防出错的检验步骤,安装文档可以一一检查你的设备是不是满足这些条件。
1.下载cuda9.0
下载地址:https://developer.nvidia.com/cuda-90-download-archive

有两种安装方式,官网推荐deb,我用了runfile。这里下载的时候,会给安装方式,建议拍照,后面会用到这行代码(手敲)。
2.安装cuda9.0
2.1 禁用nouveau驱动
$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
将以下代码写入文件:
blacklist nouveau
options nouveau modeset=0
执行
$ sudo update-initramfs –u
执行下面这行代码,如果没有输出则说明禁用成功了,如果执行完下面操作,仍有输出的话,重启一下就好

本文档详细介绍了如何在Ubuntu16.04上安装CUDA9.0、CUDNN7.0.5以及TensorFlow-GPU,包括禁用nouveau驱动、设置环境变量、验证安装成功等步骤,适合初学者参考。

1万+

被折叠的 条评论
为什么被折叠?



