tensorflow2.x 報錯 Could not load dynamic library 'cudnn64_7.dll'

前往官網下載所需得cuddn補丁：戳我

之後將

檔案全部複製到相應得

中，即可。

測試獲取gpu:

import tensorflow as tf
tf.debugging.set_log_device_placement(True)
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)
print(gpus)
logical_gpus = tf.config.experimental.list_logical_devices('GPU')
print(logical_gpus)

c = []
for gpu in logical_gpus:
    print(gpu.name)
    with tf.device(gpu.name):
        a =tf.constant([[1.0, 2.0, 3.0],[4.0, 5.0, 6.0]])
        b = tf.constant([[1.0, 2.0], [3.0, 4.0],[5.0, 6.0]])
        c.append(tf.matmul(a, b))
with tf.device('/CPU:0'):
    matmul_sum = tf.add_n(c)

print(matmul_sum)

測試結果：
```

本作品採用《CC 協議》，轉載必須註明作者和本文連結