-
Notifications
You must be signed in to change notification settings - Fork 5.9k
Closed
Labels
Description
您好,
我在使用 Conda 安装 paddlepaddle 2.0 后运行检测,结果显示
W0530 23:53:27.316645 18979 device_context.cc:346] device: 0, cuDNN Version: 7.4.
此时的主要环境为
(Ubuntu 16.04 LTS)
cudatoolkit 10.0.130 0 https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
cudnn 7.6.5 cuda10.0_0 https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
paddlepaddle-gpu 2.0.0rc0-py38_gpu_cuda10.0_many_linux
python 3.8.2 hcf32534_0 https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
可见 Conda 环境中的 cudnn 应为 7.6.5 版本,与 paddle.utils.run_check() 给出的结果不符;此版本号为我本地主机安装的 cuDNN 版本 7.4 (in /usr/local/cuda/include/cudnn.h)
更换 paddlepaddle 版本为 1.8.5,仅对 conda 虚拟环境进行如下操作:
The following packages will be DOWNGRADED:
paddlepaddle-gpu 2.0.0rc0-py38_gpu_cuda10.0_many_linux --> 1.8.5-py38_gpu_cuda10.0_many_linux
此时运行 paddle.fluid.install_check.run_check() 的输出提示 paddle 找到的 cuDNN 又变为了 conda 虚拟环境中的版本(in **/anaconda3/envs/ENV_NAME/include/cudnn.h),为
W0531 00:06:05.766619 36797 device_context.cc:260] device: 0, cuDNN Version: 7.6.
而summary_env.py 给出的环境信息如下,似乎没有找到 cuDNN
****************************************
Paddle version: 1.8.5
Paddle With CUDA: True
OS: Ubuntu 16.04
Python version: 3.8.2
CUDA version: 10.0.130
cuDNN version: None.None.None
Nvidia driver version: 418.56
****************************************
请问这一结果是什么原因造成的?paddle 会优先使用 conda 虚拟环境下的 cudnn 还是主机中安装的 cudnn?