pipenv+PyTorch(Ubuntu20.04)の環境構築に苦しんだ話

背景

• PyTorchの勉強しようと思ったので 公式ページを参考にインストール
• (何も考えてないので)pipのところをpipenvにしたらいけるでしょｗ的なノリでインストールしたらとりあえず入った
• ↓のコマンドでGPU認識するか確認する

pipenv run python -c "import torch; print(torch.cuda.is_available()); print(torch.cuda.get_device_name())"

• ↓の感じのWarningが出て頭を抱える(私はRTX3070を使ってます) github.com

解決策

pipenv installでwhlファイルのURLを指定する。

pipenv install [URL]

URLは以下画像の-f で指定してあります。URLたどると大量のwhlファイルが...！

PyTorch公式HPから

良きPyTorchライフを✋