如何安装cudnn.torch?

他们的页面 中包含安装部分,但对于不了解torch和lua的人(比如我),很难理解:该部分列举了先决条件的安装,但没有提到cudnn.torch本身。

存储库包含一堆lua文件。如何将它们用于"安装"?

点赞
用户3754413
用户3754413

你必须先安装 cuda (https://developer.nvidia.com/cuda-downloads),然后下载 cudnn 绑定(https://developer.nvidia.com/cuda-downloads)。

然后如果你已经安装了 Torch, luarocks install cudnn 应该就足够了。

要进行完整的安装,可以查看 此教程

2017-07-17 21:50:59
用户343550
用户343550

您需要从 Github 库安装:

git clone https://github.com/soumith/cudnn.torch
cd cudnn.torch
luarocks make

根据您已经安装的 CUDA 版本,您可能需要指定库的特定分支,例如 git clone https://github.com/soumith/cudnn.torch.git -b R7

2018-10-31 17:26:35