1. 程式人生 > >TensorFlow-GPU安裝前需下載的檔案

TensorFlow-GPU安裝前需下載的檔案

(1)檢查電腦naidia顯示卡匹配的cuda版本:
https://www.cnblogs.com/yingchuan-hxf/p/7710071.html
本機支援cuda 9.1,推薦安裝cuda 9.0
(2)下載cuda(需註冊)
https://developer.nvidia.com/cuda-toolkit-archive
(3)下載cudnn版本(需與cuda版本匹配,需註冊賬號)
https://developer.nvidia.com/rdp/cudnn-archive