CUDA
1. 在安装 CUDA 之前应该先知道自己本机显卡的驱动版本号, 下载 CUDA 时后面都有对应的显卡驱动版本号, 要对应好.
2. 虽然低版本的 cuda 可以安装在比它高的驱动上, 但不建议这样, 原因是在安装其他软件的 gpu 版本是冷不丁就会出问题.
比如本机的驱动是 435 的, 对应的 cuda 的最好版本可以通过 nvidia-smi 查看在右上角. 如下图所示
我本机安装的驱动是 435 的, 右上角也显示了 435 要求的最高的 cuda 版本是 10.1. 所以我也可以安装驱动低于 435 而且低于 10.1 的 cuda. 但是不建议这样安装, 会出问题. 还是按照推荐的来装.
CUDNN
安装完 cuda 就要装 cudnn,cudnn 在下载时也要注意它的版本号, 它后面也有对应的 cuda 的版本.
tensorflow 的安装
在安装 tensorflow 的 gpu 版本时需要注意下面的版本对应问题. 对 python,gcc,cudnn,cuda 都是有要求的
来源: http://www.bubuko.com/infodetail-3495199.html