tensorflow2问题

对于我近几天使用TensorFlow2出的问题做个总结:

1,是环境配置问题,我使用的是NVIDIA物理加速,就是GPU。
TensorFlow-gpu 2.0.0,CUDA10.0,cudnn7.6.5。
这仨之间的版本要一致,在TensorFlow官网查看对应CUDA的版本,再从NVIDIA官网下载对应CUDAcudnn版本。

2,CUDA目前最新版本是10.1,TensorFlow2应该是还没发布适用10.1的版本,所以会报错:

Could not load dynamic library ‘cudart64_100.dll’; dlerror: cudart64_100.dll not found

虽然把cuda/v10.1/bin下的cudart64_101改成cudart64_100可以正常使用,但我觉得还是改回v10.0版比较稳妥。

3,在使用GPU加速处理时,有时会报cudnn的错:
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
我查了半天好多说是因为版本问题,TensorFlow版本过高
也有说内存问题,修改GPU内存使用大小
在我测试下,关闭占用GPU的程序,重新运行发现错误没了,所以判断我的应该是GPU内存不够导致的,你如果在检查各版本无误时,不妨清一下GPU程序,或许就解决了。

原创文章 35 获赞 58 访问量 5890

猜你喜欢

转载自blog.csdn.net/weixin_43221560/article/details/103173008