由於換電腦了,原電腦顯卡太垃圾,跑不了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則安裝成功