如何在pytorch中更改cudnn版本

2024-09-30 06:26:49 发布

您现在位置:Python中文网/ 问答频道 /正文

使用此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容器或从源代码构建。 有人能帮我吗

非常感谢


Tags: httpsgithubcom源代码versiontorchpytorch帖子

热门问题