Win10+RTX2060安裝TensorFlow+Keras

Win10+RTX2060安裝TensorFlow+Keras

Author:Gary
Date:2019-6-8
參考資料:https://blog.csdn.net/qq_32728345/article/details/81570504

主要按照參考資料中的內容進行安裝,這裏對其中的某些坑進行補充

1.安裝顯卡驅動

英偉達官網下載驅動安裝即可,但是下載驅動後一定要在Nvidia控制面板中看一下支持的CUDA版本,目前最新版的驅動僅支持CUDA10.2,但是CUDA的下載頁面中並沒有CUDA10.2,不知道什麼情況。這種情況下需要下載老版本的驅動,我一個一個試,降了2個版本後發現支持10.1的最新驅動版本爲430.39,直接覆蓋安裝即可。

2.下載CUDA

直接進CUDA官網 https://developer.nvidia.com/cuda-toolkit-archive 下載對應的CUDA版本安裝即可,沒什麼問題。然後要按照參考資料的步驟設置環境變量,注意把有版本號的地方換成自己的CUDA版本。

3.下載cuDNN

去https://developer.nvidia.com/rdp/cudnn-archive 下載對應CUDA版本的,解壓後按參考資料複製到CUDA的目錄中即可。

3. 下載安裝Python

相信看這個教程的大部分人都安了的,大家可以放心用新版本Python,有大神爲高版本Python編譯TensorFlow

4.下載TensorFlow

TensorFlow官方更新很慢,可以到Github中下載其他人編譯的新版本,注意CUDA,cuDNN,Python版本要對應。地址:https://github.com/fo40225/tensorflow-windows-wheel

下載後解壓得到.whl文件,使用指令pip install <xxx.whl>安裝

安裝完成後可以在Python命令行中運行import tensorflow,沒有報錯即安裝成功

5. 安裝Keras和Numpy

pip install numpy
pip install keras

覺得下載很慢的童鞋可以參考我的另一個博客,簡單2行指令瘋狂加速pip安裝速度,只看方法2即可:
https://blog.csdn.net/m0_37340681/article/details/84109320

安裝完成後運行import keras,提示使用TensorFlow後端即安裝成功了

可以運行以下程序查看是否用的GPU:

import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

if __name__ == "__main__":
    print(device_lib.list_local_devices())

6. 可以開始你牛逼的深度學習之旅了

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