Skip to content

Conda 虚拟环境中 cuDNN 版本获取问题 #33208

@lytning98

Description

@lytning98

您好,

我在使用 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?

Metadata

Metadata

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions