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 协议》,转载必须注明作者和本文链接
文章!!首发于我的博客 Stray_Camel (^U^) ノ~YO。