Qiita Teams that are logged in
You are not logged in to any team

Log in to Qiita Team
Community
OrganizationAdvent CalendarQiitadon (β)
Service
Qiita JobsQiita ZineQiita Blog
Help us understand the problem. What is going on with this article?

Windows10でRTX2060を使えるようになるまで(NVIDIA DRIVER, NVIDIA CUDA ツールキット, PyTorchのインストール)

初めに

機械学習をするときに、GPUを使いたいですが、GPUを使うためには専用のソフトを入れる必要があります。具体的には以下の2つでした。

  • NVIDIA DRIVER
  • NVIDIA CUDA ツールキット
  • PyTorch

それぞれのインストールの仕方の備忘録です。

NVIDIA DRIVER

  • こちらのサイトからダウンロードします。 容量は600MBほどなのでちょっと時間がかかります。

image.png

  • 僕の環境の場合は、これらを選択しました。

  • インストールオプションでは、カスタムを選択しました。

  • NVIDIA GeForce Experienceは既にインストールされていたのでチェックを外しました。

  • クリーンインストールをしました。

NVIDIA CUDA ツールキット

  • こちらからCUDA Toolkitxx(xxはバージョン)をダウンロードします。
  • PyTorchを使う予定の方は現時点だと10.2を使ってください。
  • local版をダウンロードしました。2.5GB強あり、ダウンロードに時間がかかります。
  • こちらも、カスタムでインストール。
  • Visual Studioをインストールしていない場合は、確認が出ます。Visual Studioをインストールする予定がある場合は、個々の段階でインストールしたほうが良いみたいです。

PyTorch

こちらから自分の環境を選びます。
image.png
一番下に、コマンドが出るのでコピーしてコマンド プロンプトで実行しましょう。

windows消して、ubuntu入れようかな...

ryryry
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away