Tensorflow2.0安裝全流程-2020最新

由於換電腦了,原電腦顯卡太垃圾,跑不了GPU版本就一直沒裝,其實安裝過程蠻簡單的,只要按步驟來即可。

一、安裝Anaconda

Anaconda安裝地址:官方下載地址

這一步傻瓜式操作即可,但注意需要配置Anaconda環境

二、安裝cuda和cudnn

cudnn其實就是cuda的一個補丁

注意安裝cuda10.0的版本,tensorflow還不支持11.0以上的cuda,cudnn選擇cuda匹配的即可。

cuda下載地址:官方下載地址

cudnn下載地址:官方下載地址

  • cuda安裝傻瓜式操作,注意選擇自定義安裝,其中不要選擇以下幾項。

        CUDA-》Visual Studio Integeration / Nsight Compute

  • cuda安裝完畢後解壓cudnn

  • 將所有目錄複製到cuda目錄中,完整目錄如下

  • 配置環境變量

三、安裝Tensorflow GPU版本

pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple # 注意報錯就多裝幾次

四、測試是否安裝成功

#以下是在python環境中
import tensorflow as tf
tf.test.is_gpu_available()
# 返回結果爲True則安裝成功

 

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