CUDA ToolkitとcuDNNの概要CUDA ToolkitはNVIDIAの提供するGPGPUのためのプラットフォームです。cuDNNはNVIDIAの提供するDNNのためのライブラリです。 ailia SDKは単独でもCPUやGPU(Vulkan)を使用した推論が可能ですが、NVIDIA GPUでは、CUDA ToolkitとcuDNNを導入することで、より高速な推論が可能になります。 CUDA Toolkitのインストール下記のページのDownload Nowからインストールします。
この記事の内容 Windows環境でGPUを利用した機械学習環境を整備します。 自分のPCの設定メモですので、内容のほとんどは最後に紹介する参考サイトの焼き直しです。 CUDA Toolkitインストールの主な手順 最新のGPUドライバの入手 GPUドライバ、Tensorflow に対応した CUDA、cuDNNの入手 ドライバのインストール Tensorflow、Pytorch、CuPy のインストール システム構成 PC:DELL XPS15 2017(GeForce GTX 1050) NVIDIA GPUドライバ: 451.48(GPUドライバは最新版がよさそうです) CUDA Toolkit: CUDA Toolkit 10.0 Archive(Tensorflow) CUDA Toolkit 10.1 update2 Archive(Pytorch) cuDNN: v7.4.2
リリース、障害情報などのサービスのお知らせ
最新の人気エントリーの配信
処理を実行中です
j次のブックマーク
k前のブックマーク
lあとで読む
eコメント一覧を開く
oページを開く