tensorflow2.0穩定版GPU簡單安裝

近期很多小夥伴詢問我如何安裝GPU版本tensorflow,其實特別簡單。首先你的電腦一定要有NVIDIA顯卡,算力現在的電腦應該普遍滿足,不做考慮。
看到網上有許多GPU安裝的方法,倍感麻煩,在此做簡單安裝介紹。

第一步:下載cuda10.0和cudnn

點我獲取,提取碼:e8fu
cuda10.0安裝其實直接默認即可,當然自定義也是可以的。去掉其中的flash插件,其實無傷大雅,方便考慮,默認即可。
cudnn解壓後,複製到cuda底下,和bin相同目錄即可。


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

第二步 : 配置環境變量
如圖所示配置即可。

編輯path

第三步:tensorflow—gpu的安裝

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

pip和conda選其一安裝即可。當然配置國內鏡像會快速很多。
如此tensorflow-GPU版本的安裝就完成了。
第一次創作不足之處,請多多包涵。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章