NVIDIA CUDA toolkit on Windows - Qiitaこの記事の内容 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
