我目前负责让 tensorflow-gpu 1.8 在我的机器上运行。直到现在我一直在使用 tf-gpu 1.2，但是由于一些必需的功能，我必须升级我的安装。
在这样做之前，我想检查一下是否有最佳做法来执行此操作。我当前的设置如下所示：
正如 tf-homepage 上所写，我必须使用 CUDA v9.0 和 cuDNN v7.1。由于所有这些说明都指的是全新安装而不是更新，我不确定是否最好先卸载旧版本。
如果您已经遇到过同样的问题，请分享您的经验。谢谢！
【问题讨论】：
标签： tensorflow cuda upgrade cudnn
感谢@joão gabriel s.f.我能够成功卸载 CUDA 8.0/cuDNN 5.1 并安装最新版本的 tensorflow。由于整个过程对我来说有点混乱，我决定发布一个快速演练，也许可以帮助处于相同情况的人。
卸载
首先，我卸载了 cuda 及其所有依赖项。当我通过包管理器安装它时，我使用了 apt-get 来删除它。对于运行文件的安装，您可以查看this。
sudo apt-get --purge remove cuda
sudo apt-get autoremove
dpkg --list |grep "^rc" | cut -d " " -f 3 | xargs sudo dpkg --purge
另外，我检查了 /usr/local/ 中的任何 cuda 文件夹并将它们删除。关于cuDNN，通过删除所有cuda文件夹，对应的cuda headers和libs已经被删除。
安装
先检查显卡驱动。 CUDA 9.0 与 v384.111 驱动程序一起工作（因此不需要 390.xxx），所以我在这里无事可做。
我下载了 CUDA Toolkit 9.0 here 作为 deb（本地）。在同一个文件夹中，我执行了
dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
然后设置环境变量：
export PATH=${PATH}:/usr/local/cuda-9.0/bin
export CUDA_HOME=${CUDA_HOME}:/usr/local/cuda:/usr/local/cuda-9.0
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda-9.0/lib64
之后我按照here 的描述验证了我的安装。
我从存档中下载了 cuDNN 7.1 作为 tarball 并通过
安装了它tar -xzvf cudnn-9.0-linux-x64-v7.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h \
/usr/local/cuda/lib64/libcudnn*
启动 Python bash 后，我能够导入 tensorflow 并运行一个简单的图表。
再次感谢，祝您度过愉快的一周！
【讨论】：
见this documentation。他们说总是先从 cuda 中删除旧版本。
并且由于 cuda 9.1 需要驱动程序 >= 390 版本（检查 compatibility chart）。删除您当前的驱动程序会很好。 不过不用担心，因为 390 驱动程序在安装时附带了 cuda 9.1。
现在，作为个人建议，我想说删除与 nvidia / cuda 相关的几乎所有内容（不包括 python）。由于某些原因，在 Ubuntu 中安装和设置 CUDA 时很容易搞砸。
如果您在安装后有任何问题，请参阅ubuntu-16-04-lts-login-loop-after-updating-driver-nvidia，这是我之前回答的帖子。
【讨论】：