Ubuntu19.10安装tensorflow-gpu

前提

关于如何查看显卡型号、显卡支持的cuda版本，以及对应的tensorflow-gpu版本等的方法网上可以很容易查到，不再赘述。

安装环境为（注：英韦达官网上安装cuda9.0时，没有19.10选项，我选择的是17.04）：

1. 显卡：GeForce940MX
2. cuda: 9.0
3. cudnn: 7.0.5
4. tensorflow-gpu: 1.8.0

经过测试，可以实现显卡加速运算。

安装过程

安装过程主要参考了：

1） Ubuntu安装Tensorflow（GPU版)

2）Ubuntu 18.04安装Tensorflow(GPU)

注意：只是参考，对应版本号需要相应更改。

注意事项

1）在安装cuda时，直接执行：sudo sh cuda_9.0.176_384.81_linux.run 会报错并按安装失败，我的解决方法是：

1. 首先

sudo apt install nvidia-cuda-toolkit

2. 然后：

sudo sh cuda_9.0.176_384.81_linux.run --override

2）安装完tensorflow后，最好重启一下，或者至少在安装cudnn后要重启一下再进行接下来的步骤，不然可能会在安装完tensorflow后测试失败。

来源：CSDN

作者：梦逸清尘

链接：https://blog.csdn.net/WUDIxi/article/details/104538268