安装之前
移动端RTX30系列显卡支持cuda11.0、cudnn9.0以上版本,低版本的cuda是不兼容的,使用google版本的tensorflow需要tf2.4及以上,而bert模型是基于tensorflow1.x的,所以安装tensorflow1.x-cpu版本,tensorflow-gpu版本运行bert会报错。
具体安装步骤可以根据以下过程进行:
https://blog.csdn.net/qq_42458992/article/details/120684554
移动端RTX30系列显卡支持cuda11.0、cudnn9.0以上版本,低版本的cuda是不兼容的,使用google版本的tensorflow需要tf2.4及以上,而bert模型是基于tensorflow1.x的,所以安装tensorflow1.x-cpu版本,tensorflow-gpu版本运行bert会报错。
具体安装步骤可以根据以下过程进行:
https://blog.csdn.net/qq_42458992/article/details/120684554