深度學習搞的是如火朝天離不開各大學習框架,更少不了CUDA的計算速度加成buff。各家深度學習框架瘋狂推新,cuda也不甘落後,苦了開發者要做各種的版本選擇。一不小心就掉坑裏了,還半天爬不出來那種。
記錄下,以備再掉坑好找根繩子。
Tensorflow
Pytorch
torch-1.3 | cuda10.1 |
---|---|
torch-1.2 | cuda10.0 |
torch-1.1 | cuda9.0 |
Cuda && Driver Version
僅供參考。
深度學習搞的是如火朝天離不開各大學習框架,更少不了CUDA的計算速度加成buff。各家深度學習框架瘋狂推新,cuda也不甘落後,苦了開發者要做各種的版本選擇。一不小心就掉坑裏了,還半天爬不出來那種。
記錄下,以備再掉坑好找根繩子。
torch-1.3 | cuda10.1 |
---|---|
torch-1.2 | cuda10.0 |
torch-1.1 | cuda9.0 |
僅供參考。
01 背景介紹 GPU 目前大量應用在了愛奇藝深度學習平臺上。GPU 擁有成百上千個處理核心,能夠並行的執行大量指令,非常適合用來做深度學習相關的計算。在 CV(計
參考github案例,通過 Deep Reinforcement Learning(DQN算法)訓練能玩貪喫蛇的AI。 環境搭建 基於python + tenso