hun*_*ter 13 cuda torch cudnn
我看到许多火炬代码使用:
require cudnn require cunn require cutorch
这些包用于什么？他们与Cuda的关系是什么？
小智 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模块.
归档时间：
9 年，3 月 前
查看次数：
7393 次
最近记录：
7 年，9 月 前