Tensorflow, Pytorch版本和cuda,cudnn版本關係

深度學習搞的是如火朝天離不開各大學習框架,更少不了CUDA的計算速度加成buff。各家深度學習框架瘋狂推新,cuda也不甘落後,苦了開發者要做各種的版本選擇。一不小心就掉坑裏了,還半天爬不出來那種。 記錄下,以備再掉坑好找根繩子。 Tensorflow 參考 Tensorflow 官網推薦 Pytorch torch-1.3 cuda10.1 torch-1.2 cuda10.0 torch-1.
相關文章
相關標籤/搜索