LoginSignup
0
0

More than 1 year has passed since last update.

cudaとpytorchのインストールで気をつけるべきこと

Posted at

背景

先日cudaとpytorchの環境を導入しようと思ったのですが,全くうまくいきませんでした.
うまくいかなかった原因に気づけたので,ここでシェアできたらと思い投稿させていただきます.

目次

1.失敗したことについて
2.環境の再構築
3.おわりに

1. 失敗したことについて

環境構築の際,見様見真似で構築してしまってのが原因です

torch.cuda.is_available()

で確認したところFalse値が返ってしまい,cudaが認識されていませんでした.この時犯していた過ちが2つあり,

  • cudaの入れるバージョンが間違っていた(最新の11.3を入れていた)
  • pytorchをanacondaのGUIで入れたため互換性がめちゃくちゃだった

この2つが主な原因だと思われます.
これらを踏まえ環境を構築し直しました.

2. 環境の再構築

まずpytorchを入れることからやり直しました.
https://pytorch.org/get-started/locally/
をみてcondaのコマンドを実行しました.
パッケージインストールは全てcondaで行なったほうが良いです.互換性の問題を考えなくて済みます.

pytorchをインストールしたら次に適合したバージョン(ここでは10.2)のtoolkitをダウンロードし,インストールします.この時入れてしまったバージョンは確認しなくて良いです.

インストールが終わったらcudaのpathを確認します.

スクリーンショット 2021-05-27 123443.png

CUDA_PATHがpytorchに合ったものになっていたら成功です.
合っていなかったら書き換えてください

3. おわりに

conda install [packeage]

が超重要であることがわかりました.

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0