Windows10離線安裝tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl(對應GPU版本的tensorflow,對應python3.6)
阿新 • • 發佈:2018-11-23
Windows10離線安裝tensorflow_gpu-1.5.0rc1-cp36-cp36m-win_amd64.whl(對應GPU版本的tensorflow,對應python3.6)
1. 確保python版本是3.5 64位及以上。(tensorflow從1.2開始支援python 3.6,之前的官方是不支援的)
2. 在安裝了Anaconda之後,GPU版本的tensorflow和CPU版本的tensorflow安裝相同,在以上的基礎上(缺少的檔案安裝完畢),進入Anaconda3\Lib\site-packages\tensorflow中,執行pip install
3. 安裝CUDA(先確保顯示卡支援CUDA)
1) TensorFlow 1.6對應CUDA 9.0,去官網下載對應版本的安裝包(~1.4G)。雙擊下載的exe檔案進行安裝。
2) 新增環境變數,使具有一下4個環境變數
3) 測試一下CUDA是否安裝成功:
Cmd(命令列)輸入nvcc -V,看到版本資訊就表示安裝成功了:
4. 安裝cuDNN
TensorFlow1.6對應的版本是cuDNN7.0,去官網下載對應的安裝包(其實就是個壓縮包),將裡面的檔案解壓,然後分別新增到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0對應的三個資料夾中
5. 測試安裝成功
在cmd中輸入import tensorflow即可。或者跑個程式試一下。