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

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

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

Tensorflow

参考 Tensorflow 官网推荐
在这里插入图片描述

Pytorch

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

Cuda && Driver Version

在这里插入图片描述
仅供参考。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章