win10+cuda10.1+cudnn7.5+anaconda3 安裝 tensorflow-gpu

  •  在nvdia官網下載 cuda10.1 win10 x64, 安裝。
  • 在nvdia官網下載 cudnn-10.1-windows10-x64-v7.5.0.56.zip, 解壓,將裏面的文件夾複製到cuda的安裝目錄 比如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
  • 下載anaconda3 (2019年6月19日 Anaconda3-2019.03-Windows-x86_64.exe,注意安裝的是python3.7),一路安裝。(如果想直接在windows命令行中使用python、pip,請安裝後手動把anaconda的安裝目錄添加到環境變量path.)
  • 打開 anaconda prompt,使用pip 安裝 ,pip install tensorflow-gpu
  • 使用例子, 
    import tensorflow as tf

    此pip指令按照的是最新版本的tensorflow1.13.1(除了2的beta版本之外)。

     

    好了,總算問題來了。。。

    ImportError: DLL load failed: 找不到指定的模塊。
    
    
    Failed to load the native TensorFlow runtime.
    
    See https://www.tensorflow.org/install/errors
    
    for some common reasons and solutions.  Include the entire stack trace
    above this error message when asking for help.
    wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

    網上GitHub一查,大把人出現這個問題, Win10: ImportError: DLL load failed: 瀏覽了一遍後發現問題,“ confirmed it needs a dependency on CUBLAXX_100.DLL.
    Then I installed CUDA10.0 and it works.” cuda版本不對。。。難道要重裝cuda?將cuda降級到10.0?雖然現在cuda裝起來很容易,也不需要額外配置,可是還是麻煩啊,之前都用得好好的。。有點不甘心啊。。

tensorflow‑1.9.0‑cp37‑cp37m‑win_amd64.whl,之後在anaconda prompt 下將目錄cd到下載目錄,然後pip install xxx.whl,顯示安裝成功。

  • 測試一下:
  • import tensorflow as tf
    import numpy as np
    
    # 使用 NumPy 生成假數據(phony data), 總共 100 個點.
    x_data = np.float32(np.random.rand(2, 100)) # 隨機輸入
    y_data = np.dot([0.100, 0.200], x_data) + 0.300
    
    # 構造一個線性模型
    # 
    b = tf.Variable(tf.zeros([1]))
    W = tf.Variable(tf.random_uniform([1, 2], -1.0, 1.0))
    y = tf.matmul(W, x_data) + b
    
    # 最小化方差
    loss = tf.reduce_mean(tf.square(y - y_data))
    optimizer = tf.train.GradientDescentOptimizer(0.5)
    train = optimizer.minimize(loss)
    
    # 初始化變量
    init = tf.global_variables_initializer()
    
    # 啓動圖 (graph)
    sess = tf.Session()
    sess.run(init)
    
    # 擬合平面
    for step in range(0, 201):
        sess.run(train)
        if step % 20 == 0:
            print( step, sess.run(W), sess.run(b))

    成功輸出結果!!大功告成。

  •  如果想使用更高版本的tensorflow,可以參考網友給出的其他解決方案,或者安裝2.0beta版本。
     
- Windows 10 Home 64bit
- CUDA Toolkit 10.0 (Sept 2018)
- Download cuDNN v7.6.0 (May 20, 2019), for CUDA 10.0
- Python 3.7.3 Windows AMD64
- Tensorflow-gpu 1.13.1

wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==

---------------END-------------------------

2019年6月19日

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