Cuda 8对于Pacal架构的英伟达新一代显卡(GTX 1070/1080/Titan XP,以及未来的1080Ti)有了更加针对性的优化,对科学计算的性能提升明显。为了使用Cuda 8,我们必须使用Fedora23及以上版本的系统。
Fedora 25于2016年11月底发布,截止到本文写作时,在其上安装Cuda 8的相关资料还很少。然而通过查阅资料发现,在Fedora 25上安装Cuda意外的简单和方便,比在Ubuntu上的安装更加简单。
步骤如下:
1.清除系统上旧版本的Cuda和显示驱动。对于全新的Fedora 25系统可以跳过
sudo dnf remove xorg-x11-drv-nvidia
sudo dnf remove cuda-repo-*
sudo rm -rf /usr/local/cuda*
2.确定电脑正确连接了Cuda支持的显卡
sudo lspci | grep -i NVIDIA
3.之所以安装过程简单,是因为我们将使用negativo Nvidia repo(http://negativo17.org/nvidia-driver/)
sudo dnf config-manager --add-repo=http://negativo17.org/repos/fedora-nvidia.repo
4.安装Cuda,最新的Nvidia驱动与Cudnn(一个科学计算加速库,目前默认是v5版本,适用于Pascal架构)
sudo dnf install kernel-devel dkms-nvidia nvidia-driver-cuda
sudo dnf install cuda-devel cuda-cudnn-devel
安装时,由于需要下载驱动与Cuda安装包(1.3GB),需要花费一些时间,期间确保网络通畅
5.此时已经安装完毕。重启电脑后输入
sudo lsmod | grep nv
若显示类似于如下信息,安装即为成功:
使用 nvidia-smi 命令可以查看显卡实时工作信息。
在此之后,便可参照各科学计算/深度学习框架的安装教程进行下面的安装步骤。
需要注意的是,Cuda 8仍对gcc的版本有要求。由于Fedora 25自带gcc版本为6.1,需要手动安装4.9版本的gcc(Fedora官方软件库中只有过于古老的3.7版本gcc)。