Torch Machine Learning FrameworkでのGPU使用量の設定

Question

現在、私はTheanoを機械学習に使用しています。今はTorchを試してみたかったのです。 Theanoで

GPUのメモリ使用量のフラグを設定するオプションがあります：

os.environ["THEANO_FLAGS"] = "mode=FAST_RUN, device=gpu,floatX=float32,lib.cnmem=0.9"

のでTheanoは90％ここで、GPUの設定容量を使用しています。 Torchでは、同様のネットワークでは約30％のGPU負荷が使用されます。

Theanoに類似したTorchのGPU負荷を高く設定する方法はありますか？

Answer 1

Torchは、標準アロケータに基づいて必要なだけ多くのGPUメモリを使用します。

トーキョーの使用量は、テーラーの例のようにあらかじめ指定する必要はありません。