#背景
先日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を確認します.
CUDA_PATHがpytorchに合ったものになっていたら成功です.
合っていなかったら書き換えてください
#3. おわりに
conda install [packeage]
が超重要であることがわかりました.