我已经安装了一些传统的tensorflow软件包,以便实际上可以从物港训练这个较旧的MASK R-CNN模型。 Tensorflow-gpu == 1.13.1
我不确定是否需要指定docker映像?我知道TF2.1使用CudNN 7.6-tensorflow-gpu 1.13.1说它想要CudNN 7.4-或者如果我缺少非常简单的内容。
https://www.tensorflow.org/install/source#tested_build_configurations
我手动降低了步骤/时期和时期数,因此在CPU上运行内核大约需要10分钟。
如何使此Kaggle GPU在该内核上工作?: