[测试gpu可用报错]tensorflow2.1.0, Could not load dynamic library 'libcudart.so.10.1'

先上图在上文字:

[测试gpu可用报错]tensorflow2.1.0, Could not load dynamic library 'libcudart.so.10.1' 

Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655311: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcublas.so.10'; dlerror: libcublas.so.10: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655363: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcufft.so.10'; dlerror: libcufft.so.10: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655412: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcurand.so.10'; dlerror: libcurand.so.10: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655459: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcusolver.so.10'; dlerror: libcusolver.so.10: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655506: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libcusparse.so.10'; dlerror: libcusparse.so.10: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda-10.0/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:/data/TensorRT-6.0.1.5/lib
2020-04-16 14:22:37.655535: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libcudnn.so.7
2020-04-16 14:22:37.655545: W tensorflow/core/common_runtime/gpu/gpu_device.cc:1592] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.

 

总之就是各种文件找不到,我以为是cuda配置的问题,幸亏多留个心眼,在tensorflow1.14和tensorflow1.15上没问题,只有在tensorflow2.1.0上有问题。网上查询了下对应关系。如下图:

[测试gpu可用报错]tensorflow2.1.0, Could not load dynamic library 'libcudart.so.10.1'

我本机cuda是10.0,如果要用2.1.0,需要安装cuda10.1

解决方案:

[测试gpu可用报错]tensorflow2.1.0, Could not load dynamic library 'libcudart.so.10.1'

在tensorflow2.1.0的虚拟环境里面安装cuda10.1

 conda install cudatoolkit=10.1