美文网首页
Tensorflow-gpu==1.15.0应该装哪个版本的cu

Tensorflow-gpu==1.15.0应该装哪个版本的cu

作者: 热爱生活H_Li | 来源:发表于2021-01-07 00:46 被阅读0次

    CUDA10.0都没什么问题,最气的是亲测Tensorflow-gpu==1.15.0对应的cuDNN居然是7.6.0,我装的7.4.1居然还用不了。

    2021-01-07 01:15:48.159392: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 10156 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2080 Ti, pci bus id: 0000:02:00.0, compute capability: 7.5)
    
    2021-01-07 01:15:52.148048: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] 
    Successfully opened dynamic library libcudnn.so.7
    2021-01-07 01:15:52.866147: E tensorflow/stream_executor/cuda/cuda_dnn.cc:319]
    Loaded runtime CuDNN library: 7.4.1 but source was compiled with: 7.6.0.  
    CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. 
    If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.
    2021-01-07 01:15:52.866938: E tensorflow/stream_executor/cuda/cuda_dnn.cc:319] 
    Loaded runtime CuDNN library: 7.4.1 but source was compiled with: 7.6.0.  
    CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. 
    If using a binary install, upgrade your CuDNN library. 
    If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.
    Traceback (most recent call last):
      File "train_Sony.py", line 189, in <module>
        feed_dict={in_image: input_patch, gt_image: gt_patch, lr: learning_rate})
      File "/home/hench/anaconda3/envs/SID/lib/python2.7/site-packages/tensorflow_core/python/client/session.py", line 956, in run
        run_metadata_ptr)
      File "/home/hench/anaconda3/envs/SID/lib/python2.7/site-packages/tensorflow_core/python/client/session.py", line 1180, in _run
        feed_dict_tensor, options, run_metadata)
      File "/home/hench/anaconda3/envs/SID/lib/python2.7/site-packages/tensorflow_core/python/client/session.py", line 1359, in _do_run
        run_metadata)
      File "/home/hench/anaconda3/envs/SID/lib/python2.7/site-packages/tensorflow_core/python/client/session.py", line 1384, in _do_call
        raise type(e)(node_def, op, message)
    tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
    

    重点看第3和4个报错信息,然后你再看官网上的信息,你就知道为啥这么气人了。


    图片.png

    所以你说这不是搞笑么。没办法,只能卸载cudnn7.4.1然后重新装cudnn7.6.0。

    相关文章

      网友评论

          本文标题:Tensorflow-gpu==1.15.0应该装哪个版本的cu

          本文链接:https://www.haomeiwen.com/subject/odeaoktx.html