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

2021-01-07  本文已影响0人  热爱生活H_Li

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。

上一篇 下一篇

猜你喜欢

热点阅读