TensorFlow知识图谱实战
上QQ阅读APP看书,第一时间看更新

1.5.1 10/20/30系列显卡选择的GPU版本

目前市场上同时有10、20、30系列显卡,对于需要调用专用编译器的TensorFlow来说,不同的显卡需要安装不同的依赖计算包。不同显卡的TensorFlow版本以及CUDA和cuDNN的对应关系如表1.1所示。请读者注意根据自己的配置选择不同的TensorFlow版本。

表1.1 10/20/30系列显卡的版本对比

注意

当显卡为20系列或者以前的显卡时能够使用的最高版本为2.3.x,当显卡为30系列时最低的TensorFlow版本为2.4.x,版本不符时,则无法调用相应的计算库启动TensorFlow的计算任务。

下面以TensorFlow 2.4为例演示完整的CUDA和cuDNN的安装步骤,不同的版本安装过程基本一致。