1. 程式人生 > >win10+Anaconda中安裝tensorflow(gpu版)

win10+Anaconda中安裝tensorflow(gpu版)

我之前已經在Anaconda中安裝了cpu版的tensorflow,但由於深度學習要涉及到大量的計算,cpu實在太慢。這時gpu版的tensorflow就派上大用場了。所有相關環境的下載路徑都在參考博文中,安裝步驟也是跟著博文一起走的,只不過我在新增幾條注意事項。

注意:

1.在下圖步驟中:

這個名字時可以隨便起的,因為我自己以前安裝cpu版的tensorflow,所以gpu版的conda環境我起名為tf-gpu.通過以下程式碼

conda create -n tf-gpu pip python=3.5

下面是我建好的gpu的conda環境

2 .當cuda安好之後,檢測是否安好方法如下:

3.安裝cudnn的時候,我是跟著下面這個博主的文章進行的。(博文1的方法我沒試,2肯定沒問題,有興趣的同學可以試試1的方法)

4.最後測試