close

上網一直爬文, 每次都裝失敗

最後莫名把tensorflow GPU版裝好了! 應該吧!

首先把原本的tensorflow GPU移除

pip uninstall tensorflow-gpu

未命名.png

pip install tensorflow-gpu==1.5

未命名.png

然後下載這些東西

https://developer.nvidia.com/cuda-90-download-archive

https://developer.nvidia.com/rdp/cudnn-archive

擷取.PNG

未命名.png

先把cuda_9.0.176_win10.exe安裝好, 隨便一個資料夾

未命名.png

然後把解好壓縮的cudnn-9.0-windows10-x64-v7裡面檔案複製出來, 到上面的圖裡面

未命名.png

然後把

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

加到環境變數

回到Anaconda Prompt

輸入Python

import tensorflow as tf

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

未命名.png

居然跑成功了, 痛哭涕零.........

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

arrow
arrow
    全站熱搜
    創作者介紹
    創作者 張郎屋 的頭像
    張郎屋

    張郎生活的筆記

    張郎屋 發表在 痞客邦 留言(0) 人氣()