使用此https://gist.github.com/kmhofmann/cee7c0053da8cc09d62d74a6a4c1c5e4#installing-cudnn安装cudnn时
我下载并安装了cudnn 8.2.1
但是在安装PyTorch并使用torch.backends.cudnn.version()
之后,它显示8002
我在PyTorch上找到一个帖子,上面写着:https://discuss.pytorch.org/t/how-to-change-cudnn-version-in-pytorch/75625
安装日志显示使用了哪个CUDA运行时和cudnn(在文件名中)。 如果您想使用本地库,则必须从源代码构建。
但我不知道如何使用NGC容器或从源代码构建。 有人能帮我吗
非常感谢
目前没有回答
相关问题 更多 >
编程相关推荐