1. 程式人生 > >使用conda快速在windows下安裝GPU版本的Tensorflow+CUDA+cudnn

使用conda快速在windows下安裝GPU版本的Tensorflow+CUDA+cudnn

之前我嘗試過很多種方法,首先要找python對應版本的tf,又要找tf對應版本的cuda,最後還要找cuda對應版本的cudnn,麻煩事一大堆,還不一定能一次安排上,偶然間發現了conda的強大,來跟大家分享一下。

首先推薦安裝Anaconda,有多好用就不多說了,基本上有常用的各種包,關鍵是還有大殺器conda,從此以後就告別pip了。

首先,確認你的python版本是3.6(反正不能是3.7),如果要快速遷移python版本,直接在命令列下輸入:

conda install python==3.6

然後直接:

conda install tensorflow-gpu

conda會自動幫你下載gpu版本的tensorflow所需的所有環境,包括cuda和cudnn,我忘了截圖,印象中tf目前最新版本1.12.0對應的是cuda9.0.1和cudnn7.1.4,但是這都不用自己操心,conda自動識別下載,默默等待下載安裝完成就好了,親測可用。