火炬中'cuda''cunn''cunn'和'cutorch'之间有什么区别和关系?

时间:2016-07-05 03:02:12

标签: cuda torch cudnn

我看到许多火炬代码使用:

require cudnn
require cunn
require cutorch

这些包用于什么?他们与Cuda的关系是什么?

2 个答案:

答案 0 :(得分:10)

所有3个用于torch7的CUDA GPU实现。

cutorch是torch7的cuda后端,为火炬中的CUDA实现提供各种支持,例如GPU内存中的张量的CudaTensor。在与GPU交互时还添加了一些有用的功能。

cunn在nn库上提供了额外的模块,主要是透明地将这些nn模块转换为GPU CUDA版本。这样可以很容易地通过cuda将神经网络切换到GPU,反之亦然!

cuDNN是NVIDIA的cuDNN库的包装器,它是CUDA的优化库,包含各种快速GPU实现,例如卷积网络和RNN模块。

答案 1 :(得分:1)

不确定' cutorch'不过是我的理解:

Cuda:使用GPU的库。

cudnn:在GPU上做神经网络的库(可能使用Cuda与GPU对话)

来源:https://www.quora.com/What-is-CUDA-and-cuDNN