上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的安装步骤,不同的版本安装过程基本一致。