weixin_43518083的博客

原创 TensorFlow-gpu 2.3.1安装

CUDA-11.0 + cuDNN-7.6 + TensorFlow-2.1.0 TensorFlow-1.8.0下，用ResNet-50跑10800张RGB图片，一个epoch用时2900s左右，后发现只用cpu在跑，未启动gpu。 然后打算下载相应版本的CUDA和cuDNN，发现Linux已经安装配置好CUDA-11.0和cuDNN-7.6。 将TensorFlow更新到对应的版本 TensorFlow-2.1.0即可，一个epoch时间减少到900s （TensorFlow-1.5以上不再区分Tens