logo
Loading...

我的tensorflow-gpu 安装成功了吗? - Cupoy

我的tensorflow-gpu 安装成功了吗?

2020/05/24 下午 07:54
電腦視覺深度學習討論版
骑士
觀看數:49
回答數:5
收藏數:0


回答列表

  • 2020/05/24 下午 08:00
    骑士
    贊同數:0
    不贊同數:0
    留言數:0

    不好意思,我补充一下我的问题。

    我按照网络上的教程安装了多个版本的tensorflow-gpu和tensorflow的虚拟环境,导致我不清楚是否已经成功使用上tensorflow-gpu。请帮忙按照我正文中的截图内容来判定一下是否有问题?真的弄了好久,如果有问题能否提供新的参考资料让我重新安装一下?非常感谢

  • 2020/05/25 下午 00:41
    Jeffrey
    贊同數:0
    不贊同數:0
    留言數:0

    Hello, 從訊息來看, CUDA 安裝沒有成功, Tensorflow GPU 有安裝但是抓不到GPU.

    有關 CUDA, 有些步驟:

    1.  CUDA 安裝

    要確認一下, 顯卡能支持

    2. cuDNN

    請根據你的CUDA版本選擇對應版本下載!

    解壓縮後,將三個資料夾中的檔案,個別複製到CUDA的相對應資料夾中。

    cuDNN下載:https://developer.nvidia.com/cuda-toolkit-archive

    3. 微軟的 Vitual studio


    安裝 tensorflow-gpu

    https://tensorflow.google.cn/install/gpu?hl=zh-cn

    必須在系統中安裝以下 NVIDIA® 軟體:


  • 2020/05/26 下午 09:49
    骑士
    贊同數:0
    不贊同數:0
    留言數:0

    谢谢您的整理回答:不过我已经按照好了NVIDIA® GPU 驅動程式CUDA® 工具套件以及cuDNN SDK。如下截图所示。另外您提供的Tensorflow GPU的安装是限Linux系统的(我的是window系统)。请问还有其它解决方法吗,谢谢?

  • 2020/05/26 下午 10:16
    骑士
    贊同數:0
    不贊同數:0
    留言數:0

    刚刚又用各种方式测试了一些结果,如下,请问老师这是什么原因?

    显示没有使用GPU的结果如下:

    显示有使用GPU的结果如下

  • 2020/05/27 下午 05:12
    Jeffrey
    贊同數:0
    不贊同數:0
    留言數:3

    建議把 GPU device mapping 到 cell 裡面, 參考如下:


    import os

    os.environ["CUDA_VISIBLE_DEVICES"] = "0"

    # os.getcwd()   

    import tensorflow as tf

    W = tf.constant([1.0, 2.0, 3.0, 4.0], shape=[2, 2], name='W')

    x = tf.constant([1.3, 2.4], shape=[2, 1], name='x')

    y = tf.matmul(W, x)

    # 只使用 30% 的 GPU 記憶體

    gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction = 0.3)

    # Creates a session with log_device_placement set to True. 

    sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True)) 


    PS: 若有問題, 請把 tensorflow 移除, 只保留tensorflow-gpu