上传者: 38747815
|
上传时间: 2021-11-05 02:58:16
|
文件大小: 45KB
|
文件类型: -
之前的文章讲过用Tensorflow的object detection api训练MobileNetV2-SSDLite,然后发现训练的时候没有利用到GPU,反而CPU占用率贼高(可能会有Could not dlopen library ‘libcudart.so.10.0’之类的警告)。经调查应该是Tensorflow的GPU版本跟服务器所用的cuda及cudnn版本不匹配引起的。知道问题所在之后就好办了。
检查cuda和cudnn版本
首先查看cuda版本:
cat /usr/local/cuda/version.txt
以及cudnn版本:
cat /usr/local/cuda