1、背景
torch和tensorflow两大主流的深度学习训练框架(tensorflow框架现在似乎开始没落了)。
参考torch官网:
https://pytorch.org/get-started/locally/
参考tensorflow官网:
https://tensorflow.google.cn/install/source#gpu
就是这么不和谐。
比如, tensorflow推荐CUDA 10.1,torch推荐10.2;
比如, tensorflow推荐CUDA 11.0,torch推荐11.1;
关于tensorflow,真的要按推荐的CUDA,以及cuDNN来匹配,否则是用不了GPU的,可以参考博客: