Google Colab升/降级TensorFlow以及cudnn的正确姿势
Google Colab升/降级TensorFlow以及cudnn的正确姿势
背景
跑leaf-audio论文的时候,代码用到了lingvo库,当前lingvo库的最新版本是0.10.0,并且与tensorflow-gpu版本强依赖(2.6,而colab是2.7)。
直接按要求pip tensorflow-gpu 2.6后,又发现cudnn版本不对。也就是说,tensorflow 2.6版本是基于cudnn 8.1编译的,需要8.1以上的cudnn才行。但是系统里面的cudnn是8.0.5。
经过n天的摸索
原创
2021-11-26 00:27:26 ·
2440 阅读 ·
3 评论