Skip to content

GPUで動作する実行ファイルと評価パラメータのセット

Compare
Choose a tag to compare
@SakodaShintaro SakodaShintaro released this 12 May 08:02
· 1189 commits to master since this release

Windows10, CUDA9.2, cuDNN7.5.1, libtorch1.1でビルドしたものです。dllは含めていないので各種dllが必要です。

・Nvidiaドライバ
https://www.nvidia.co.jp/Download/driverResults.aspx/147226/jp

・CUDA9.2
https://developer.nvidia.com/cuda-92-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

・cuDNN7.5.1(入手には登録が必要です)
https://developer.nvidia.com/rdp/cudnn-download

・libtorch
https://pytorch.org/get-started/locally/
ダウンロードして解凍した後、環境変数Pathにlibtorch/libへのパスを追加してください。