1)

最近在使用tensorflow-gpu 2.2.0版本时遇到了

Could not load dynamic library 'libcudart.so.10.1'; 

Could not load dynamic library 'libcublas.so.10';

Could not load dynamic library 'libcurand.so.10';

....

解决方法是安装与tensorflow-gpu版本对应的CUDA

(建议使用anacoda虚拟环境)

conda install cudatoolkit==10.1

(tensorflow-gpu 2.2.0与cuda10.1对应)

 

2)

之后新的报错:

Could not load dynamic library 'libcudnn.so.7'; dlerror: libcudnn.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda/lib64:/share/local/slocal/lib:/usr/local/cuda/lib64:/share/local/lib:$LD_LIBRARY_PATH

解决办法是安装cudann

conda install cudnn -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

 

3)

除此之外还要注意tensorflow-gpu2.2.0要使用python3.5以上的版本

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐