您的位置: 首页 > 文章 > Tensorflow, Pytorch版本和cuda,cudnn版本关系 Tensorflow, Pytorch版本和cuda,cudnn版本关系 分类: 文章 • 2024-01-22 20:32:46 深度学习搞的是如火朝天离不开各大学习框架,更少不了CUDA的计算速度加成buff。各家深度学习框架疯狂推新,cuda也不甘落后,苦了开发者要做各种的版本选择。一不小心就掉坑里了,还半天爬不出来那种。 记录下,以备再掉坑好找根绳子。 Tensorflow 参考 Tensorflow 官网推荐 Pytorch torch-1.3 cuda10.1 torch-1.2 cuda10.0 torch-1.1 cuda9.0 仅供参考。