1、准备
-
请先看好各种软件的版本对应要求,这仨一定要对应好。
Tensorflow不同版本要求与CUDA及CUDNN版本对应关系 -
知道要下哪些版本了,就预先做好各种软件下载工作。
首先下载好英伟达的驱动 NVIDIA驱动下载
注意!!!下载好跟自己显卡对应的驱动。显卡的产品类型、系列那些如果之前已经装好了驱动,则可以通过命令 nvidia-smi查询到。没有装刚买来就自己查。
即使你的机器之前已经装过驱动,那也最好重新装一遍驱动,因为那个CUDA一定要对应起来。不然后面有坑!下载CUDA,链接 cuda-toolkit-archive
请注意这里一定要选择下载runfilw文件,不是deb!,不然会覆盖之前的显卡驱动带来问题。
最后下载cuDNN,cuDNN下载地址,我下的7.0.5版本
2、安装驱动
2.1、正常装驱动。
按ctrl+alt+f2(有的是f1)进入字符界面命令行,先删除以前的驱动:
sudo apt-get purge nvidia*
sudo apt-get a