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
18
Help us understand the problem. What is going on with this article?
@mml

Ubuntu 20.04LTS + GPU に cuda, Anaconda, Tensorflow, Pytorch を入れる

書くこと無いくらいに簡単に入りましたが、一応メモっときます。

手順

  1. Ubuntu 20.04LTS デスクトップをインストールする。
    https://jp.ubuntu.com/download
    インストール後の処理をして、ソフトウェアのアップデートが終わるのを待つ。

  2. Ubuntu デスクトップにログインして、左下のメニューを開くと、中に「ソフトウェアアップデート」というメニューがあるのでそれを開く。

  3. 開いたウィンドウの中に「追加のドライバ」というタブがある。GPU が認識されていれば「nvidiaのドライバを使う」という項目がいくつかあって、バージョンが選べる。適切なバージョン (GTX/RTX 系なら 440 で良いはず) を選んで「変更を適用」を押す。これで、待っていればインストールされる。

  4. ターミナルを開いて、下記コマンドを打つ。これで cuda が入る。

$ sudo apt update
$ sudo apt install nvidia-cuda-toolkit

2020/05/07 時点では cuda 10.1 が入る。違うバージョンを入れることもできる(みたい)。

  1. Anaconda をダウンロードする。 https://www.anaconda.com/

Get Start -> Install Anaconda Individual Edition -> Download -> Linux Python 3.7 64-Bit (x86) Installer を選んでダウンロード。

  1. ダウンロードしたフォルダで下記コマンドを実行 (ファイル名はバージョンによって微妙にかわってるかもしれない)
$ bash Anaconda3-2020.02-Linux-x86_64.sh

必要に応じて conda create -n tf などする。

  1. tensorflow-gpu をインストール
$ conda install tensorflow-gpu
  1. pytorch をインストールする。
$ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

これで、GTX 2080, RTX 2080, GTX1660 あたりでは問題なく動きました。あとで、もうちょっと加筆します。

18
Help us understand the problem. What is going on with this article?
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
mml
LEDを光らせる屋内での軽作業に従事して早300年。LEDに関係しそうな技術には、だいたい何でも手を出します。

Comments

No comments
Sign up for free and join this conversation.
Sign Up
If you already have a Qiita account Login
18
Help us understand the problem. What is going on with this article?