近期很多小夥伴詢問我如何安裝GPU版本tensorflow,其實特别簡單。首先你的電腦一定要有NVIDIA顯示卡,算力現在的電腦應該普遍滿足,不做考慮。
看到網上有許多GPU安裝的方法,倍感麻煩,在此做簡單安裝介紹。
第一步:下載下傳cuda10.0和cudnn
點我擷取,提取碼:e8fu
cuda10.0安裝其實直接預設即可,當然自定義也是可以的。去掉其中的flash插件,其實無傷大雅,友善考慮,預設即可。
cudnn解壓後,複制到cuda底下,和bin相同目錄即可。
這樣就完成了重要的一步。
第二步 : 配置環境變量
如圖所示配置即可。
第三步:tensorflow—gpu的安裝
pip install tensorflow-gpu==2.0.0
conda install tensorflow-gpu=2.0.0
pip和conda選其一安裝即可。當然配置國内鏡像會快速很多。
如此tensorflow-GPU版本的安裝就完成了。
第一次創作不足之處,請多多包涵。