深度学习初学者使用TensorFlow框架还是pytorch框架更合适呢

Question

就目前而言，深度学习初学者使用TensorFlow框架还是pytorch框架更合适呢？

Answer 1

pytorch，如果有折腾的能力就去tf。
原因就是pytorch几年前的代码，你用新版本跑基本上问题不大（只要不用torch2.x的去跑1.x的代码，改动不会很大），最多改几个numpy或者弃用的api，cuda兼容也好。tf那就不一样了，30系和40系的新显卡你想跑tf1.x的代码，cuda都能卡你很久（这两个要cuda11.0，tf1.x的版本基本上最高支持到cuda10.2,有些方法可以在cuda11.0下面跑，但是你要自己折腾还不一定成功），更不用说配套的环境，某些库小版本不一样都能有莫名其妙的错误等着你