初めに
機械学習をするときに、GPUを使いたいですが、GPUを使うためには専用のソフトを入れる必要があります。具体的には以下の２つでした。
- NVIDIA DRIVER
- NVIDIA CUDA ツールキット
- PyTorch
それぞれのインストールの仕方の備忘録です。
NVIDIA DRIVER
-
こちらのサイトからダウンロードします。
容量は600MBほどなのでちょっと時間がかかります。
-
僕の環境の場合は、これらを選択しました。
-
インストールオプションでは、カスタムを選択しました。
-
NVIDIA GeForce Experienceは既にインストールされていたのでチェックを外しました。
-
クリーンインストールをしました。
NVIDIA CUDA ツールキット
- こちらからCUDA Toolkitxx（xxはバージョン）をダウンロードします。
- PyTorchを使う予定の方は現時点だと10.2を使ってください。
- local版をダウンロードしました。2.5GB強あり、ダウンロードに時間がかかります。
- こちらも、カスタムでインストール。
- Visual Studioをインストールしていない場合は、確認が出ます。Visual Studioをインストールする予定がある場合は、個々の段階でインストールしたほうが良いみたいです。
PyTorch
こちらから自分の環境を選びます。
一番下に、コマンドが出るのでコピーしてコマンド プロンプトで実行しましょう。
windows消して、ubuntu入れようかな...