在GTX1080显卡上,Ubuntu 16.04下安装CUDA 8.0问题

Q:我在Ubuntu 16.04下安装CUDA 8.0,安装的过程与安装手册相同,先卸载Ubuntu自带驱动,然后安装从官方网站上下载的驱动。等我把CUDA和cuDNN都安装好之后,发现运行Theano 和TensorFlow会出现找不到GPU的提示。但是我通过nvidia-smi查看设备信息是可以找到这些GPU的。我使用的是GTX 1080显卡。

A:应该是驱动问题。在NVIDIA官网下载的16.04 Debian安装包的驱动版本是361,而应该安装的版本是367(367.18)。通过更新驱动,Theano 和TensorFlow就都可以使用了。

在此列出安装的步骤:

Steps:

1、下载cuda-8.run file和Nvidia-367.18-driver.run file

2、删除系统自带驱动apt remove --purge nvidia*,删除系统自带cuda文件aptremove --purge libcuda*(如果有),删除nvidia.deb包,删除文件夹/usr/local/cuda*(如果有)

3、重启。在这个过程中显示可能会不正常,所以为了避免麻烦可以通过tty或ssh进行操作。

4、运行cuda-8.run文件安装。然后重启。

5、运行Nvidia-367.18.run文件安装。然后重启。

SOGA,非常好

谢谢哦,希望对你有用。