web-dev-qa-db-ja.com

トーチの「cuda」「cudnn」「cunn」と「cutorch」の違いと関係は何ですか?

多くのトーチコードが使用されているのがわかります。

require cudnn
require cunn
require cutorch

これらのパッケージは何に使用されますか?彼らとCudaとの関係は何ですか?

13
hunter

3つすべてがtorch7のCUDAGPU実装に使用されます。

cutorch はtorch7のcudaバックエンドであり、GPUメモリのテンソル用のCudaTensorなど、トーチでのCUDA実装のさまざまなサポートを提供します。また、GPUと対話するときにいくつかの便利な機能を追加します。

cunn nnライブラリに追加のモジュールを提供し、主にそれらのnnモジュールをGPUCUDAバージョンに透過的に変換します。これにより、cudaを介してニューラルネットワークをGPUに、またはその逆に簡単に切り替えることができます。

cuDNNは、NVIDIAのcuDNNライブラリのラッパーです。これは、畳み込みネットワークやRNNモジュールなど、さまざまな高速GPU実装を含むCUDA用に最適化されたライブラリです。

10
SeanNaren

'cutorch'が何であるかはわかりませんが、私の理解からです。

Cuda:GPUを使用するためのライブラリ。

cudnn:GPUでニューラルネット関連の処理を行うライブラリ(おそらくCudaを使用してGPUと通信します)

ソース: https://www.quora.com/What-is-CUDA-and-cuDNN

1
Charlie Parker