今回はpytorchをubuntu(16.04)で使用するのに行ったことに関して備忘録として残しておきます。細かな技術的なことに関して深堀できておらず「とりあえず動けばいい」という姿勢なのでversionが違う場合や本設定で動作を保証する記事ではないのでご了承ください。全て独学なので認識が間違っているところがあるかもしれませんのでその点もご了承ください。1.0)でモデルを構築しようとすると下記のようなエラーが出ました。..\aten\src\THC\THCGeneral.cpp:74yaakuCUDAって何...?というレベルだったのでググると「GPUにおいて並列処理を行うプログラミングモデル」ということで、どうやらNVIDIAの製のGPUを用いる際に効率的に並列処理を行ってくれるライブラリーのようなもののようです。learning用のライブラリもインストールする必要があり、CUDAとcuDNNとpytorchのversion違いで互換性がないものがあり、組み合わせを正しく選ばないとエラーが出るのが今回のエラーの原因のようでした。yaakuツライ...どのバージョンをインストールしたか?結論から書くと下記バージョンで動作を確認しました。90pytorchの公式に上記設定でインストールするコードがあったので載せておきます。pytorchほんとはバージョンごとの組み合わせによる動作有無を調べれば親切かと思いますが、データ分析をやる上では動けばよいので必要に応じて調べます...。互換性は問題ないですが、cuDNNに関しては使用するNVIDIAのGPUによってインストールするバージョンが違うようです。cuDNNのバージョン確認に関して次はcuDNNのバージョンの確認を行います。私はGCPを使用しているため使用するGPUはVMインスタンス詳細から調べられます。P100というタイプのものを使用しているようです。この情報をもとにNVIDIAの公式からドライバーの情報を確認します。384.183のドライバーが必要ということがわかりました。devicesこのドライバーをインストールするためにubuntuのドライバーのリポジトリを参照します。リポジトリの登録後、必要なドライバーをインストールします。nvidia-384以上でCUDA、pytorch、cudNNのバージョンの互換性が取れたものが揃ったかと思います。次に動作確認を行いますが、インストール後再起動する必要があるようですのでご注意ください。reboot動作確認を行うちゃんと動くのか確認を下記のコードで行います。businessこれで何かしら出力でたら設定終了です。株のシステムトレードにハマり、今はkaggleにハマっています。kaggleでメダルを複数取得後、現在はデータサイエンティストとして働いています。1から色々勉強した内容を残すために開設しました。シストレとkaggleの。