天天看點

tensorflow2.0穩定版GPU簡單安裝

近期很多小夥伴詢問我如何安裝GPU版本tensorflow,其實特别簡單。首先你的電腦一定要有NVIDIA顯示卡,算力現在的電腦應該普遍滿足,不做考慮。

看到網上有許多GPU安裝的方法,倍感麻煩,在此做簡單安裝介紹。

第一步:下載下傳cuda10.0和cudnn

點我擷取,提取碼:e8fu

cuda10.0安裝其實直接預設即可,當然自定義也是可以的。去掉其中的flash插件,其實無傷大雅,友善考慮,預設即可。

cudnn解壓後,複制到cuda底下,和bin相同目錄即可。

tensorflow2.0穩定版GPU簡單安裝

這樣就完成了重要的一步。

第二步 : 配置環境變量

如圖所示配置即可。

tensorflow2.0穩定版GPU簡單安裝

第三步:tensorflow—gpu的安裝

pip install tensorflow-gpu==2.0.0
           
conda install tensorflow-gpu=2.0.0
           

pip和conda選其一安裝即可。當然配置國内鏡像會快速很多。

如此tensorflow-GPU版本的安裝就完成了。

第一次創作不足之處,請多多包涵。

繼續閱讀