Tensorflow, Pytorch版本和cuda,cudnn版本关系

深度学习搞的是如火朝天离不开各大学习框架,更少不了CUDA的计算速度加成buff。各家深度学习框架疯狂推新,cuda也不甘落后,苦了开发者要做各种的版本选择。一不小心就掉坑里了,还半天爬不出来那种。

记录下,以备再掉坑好找根绳子。

Tensorflow

参考 Tensorflow 官网推荐
Tensorflow, Pytorch版本和cuda,cudnn版本关系

Pytorch

torch-1.3 cuda10.1
torch-1.2 cuda10.0
torch-1.1 cuda9.0

仅供参考。