针对上述bug 通过检查是安装的cudnn和cudatoolkit的版本不对造成的。
通过更新cudatoolkit=10.0.130 和cudnn=7.6.5版本后,上述bug就解决了
安装方法如下所示:
conda install cudatoolkit=10.0.130 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/