参考pyenvのインストールanacondaのインストールanacondaの仮想環境を構築PyTorch(1.2.0)のインストールcondaの場合pipの場合(pip=pip3の場合)PyTorch(1.0.1)のインストールPyTorchの動作確認PyTorchに関する確認cudaに関する確認おわりにはじめにこんにちは、がんがんです。インターンにてPyTorchの実験を行う開発環境を整える必要がありました。そのため、ブックマークにてまとめていたものを備忘録として改めてまとめておきこうと思います。1.0.1に関しても書いてます)参考https://medium.com/@tacchan7412/ubuntu-gtx1080-cuda-pytorch%E3%81%AE%E7%92%B0%E5%A2%83%E3%82%92%E4%B8%80%E6%B0%97%E3%81%AB%E6%95%B4%E3%81%88%E3%81%9F%E8%A9%B1-a983746e840https://qiita.com/yukoba/items/4733e8602fa4acabcc35https://qiita.com/koh_0227/items/cabf4dce9d8e39a5b4d4pyenvのインストール環境を汚してしまうとあとで死に目にあうため、まずはpyenv環境を構築します。export先は.bashrcとするため、nanoコマンドから直接書き込んでも大丈夫です(参照が一部zshなので注意です)。~/.bashrcpyenvのインストールが成功しているかを以下の確認してみましょう。--versionanacondaのインストールPython3.7とanacondaをインストールしていきます。必要なパッケージを先にインストールします。パッケージをインストールせずとも上手くいくこともありますが入れておいた方が念のためです。libpng-devanacondaのバージョンを確認してインストールします。anaconda3-2019.07Python3.7も一緒にインストールします。3.7.4anacondaの仮想環境を構築condaの仮想環境は以下のコマンドで作成することが可能です。anaconda今回作成したいのは以下の名前です。anacondaPyTorch(1.2.0)のインストールPyTorchのインストールは公式ページからいけます。condaから行う場合、pipから行う場合のどちらも記載しておきます。torchvisionPyTorch(1.0.1)のインストール最新のPyTorchをインストールした場合、上手く動作しないことがあります。私の場合、TensorboardXの実行時にエラーがおき、原因はPyTorchのバージョンによるものである可能性が高かったです。TensorboardXのissuesについては貼っておきます。https://github.com/lanpa/tensorboardX/issues/483この場合、.whlファイルを直接インストールする必要があります。上記の環境でPyTorch=1.0.1をインストールする場合は以下のコマンドです。https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp37-cp37m-linux_x86_64.whl公式ではtorchvisionについてバージョン指定がありません。こちらについてもバージョンを合わせておくと後のエラーを防ぐことが可能です。実際にインストール出来ているかを確認していきます。print(x)cudaに関する確認cudaに関する確認コマンドは以下の通りです。torch.cuda.get_device_name(0)おわりに今回はUbuntu環境にてPyTorchの実行環境を構築しました。現在は研究室の環境でも同様の環境を作成しているところです。この備忘録を参考にしながら楽しんで環境を構築していこうと思います。»プロフィールid:gangannikki趣味や研究の備忘録として使用しています。メモ記事もありますがお許しください。1日1つの学びがあればいい。IoTなどなど書いてます。(5)はてなブログをはじめよう!gangannikkiさんは、はてなブログを使っています。ブログを報告するスターをつけました引用をストックしましたストック一覧を見る閉じる引用するにはまずログインしてくださいログイン閉じる引用をストックできませんでした。再度お試しください閉じる限定公開記事のため引用できません。