转载自:https://blog.csdn.net/Weyoung_/article/details/108853972
最近在学习百度的paddle框架时需要CUDNN7.6以上的版本,但是服务器现有的是7.0,测试时会报错如下:
W0928 16:48:54.643898 47177 device_context.h:155] WARNING: device: 0. The installed Paddle is compiled with CUDNN 7.6, but CUDNN version in your machine is 7.0, which may cause serious incompatible bug. Please recompile or reinstall Paddle with compatible CUDNN version.
需要修改cudnn的版本,但是服务器同时还有别的程序,希望不影响其他程序,只修改当前终端的cudnn版本,操作如下:
- https://developer.nvidia.com/rdp/cudnn-archive,去该网站下载cudnn7.6.3 for cuda9.0。得到.solitairetheme8的后缀,没关系,手动改成.taz。然后移动到/home/xxx/cudnn/cudnn7.6.4-cuda9.0/下,这个文件夹是自己创建的,担心和别的cudnn搞混了。解压这个文件 tar -xvf cudnn-9.0-linux-x64-v7.6.4.38.tgz,得到cuda目录。
- 修改当前终端的环境变量,重新指定CUDNN的目录为7.6.4版本
-
export CUDNN_HOME= /home/xxx /cudnn/cudnn7. 6.4-cuda9. 0/cuda
-
export LD_LIBRARY_PATH=${CUDNN_HOME}/lib64:$LD_LIBRARY_PATH
-
export CPLUS_INCLUDE_PATH=${CUDNN_HOME}/include:$CPLUS_INCLUDE_PATH
这样就完成了,执行
-
python tools/infer.py -c configs/ppyolo/ppyolo.yml -o use_gpu=true weights=https://paddlemodels.bj.bcebos.com/object_detection/ppyolo.pdparams --infer_img=demo/000000014439_640x640.jpg
得到下述说明,说明修改成功
注意:上面的方法只临时修改了当前终端的环境变量,不影响其他终端。每次打开新的终端都需要执行一次修改环境变量的操作。