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

在這裏插入圖片描述
僅供參考。

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