关于安装cuda与cudnn以及tensorflow的一些总结

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都是有要求的

扫描二维码关注公众号,回复: 10539408 查看本文章

猜你喜欢

转载自www.cnblogs.com/answerThe/p/12643530.html
今日推荐