Ubuntu18.04とRTX3090搭載のマシンにpyenvをインストールし、pytorchを動かす方法

環境

OS : Ubuntu 18.04
GPU : RTX3090
CUDA : 11.1
cuDNN : 8.04

含まれる内容

• 起こったこと
• 順次インストールと対応方法の記録

起こったこと

以下の2点でつまづいた。

• pyenvへのpythonのインストール時のbuildエラー
• pytorch実行時のエラー

順次インストールと対応方法の記録

pyenvのインストールや上記それぞれのエラー対応方法は以下の通り。

###pyenv のインストール

$ git clone https://github.com/yyuu/pyenv.git ~/.pyenv

vimを使ってbash_profileを編集する
bash_profileが何か、については以下を参照
.bash_profile ? .bashrc ? いろいろあるけどこいつらなにもの？

$ vim ~/.bash_profile

以下を追記

export PYENV_ROOT=$HOME/.pyenv
export PATH=$PYENV_ROOT/bin:$PATH
eval "$(pyenv init -)"

以下でbash_profileの保存

$ source ~/.bash_profile

正しくをインストールされているかを確認するには

$ which pyenv

を実行し、

/home/{USERNAME}/.pyenv/bin/pyenv

のようにかえってくれば正しくインストールされている

###pyenv-virtualenv のインストール

$ git clone https://github.com/yyuu/pyenv-virtualenv.git ~/.pyenv/plugins/pyenv-virtualenv

vimを使ってbash_profileを編集する

$ vim ~/.bash_profile

以下を追記

eval "$(pyenv virtualenv-init -)"

以下でbash_profileの保存

$ source ~/.bash_profile

###pyenvへのpythonのインストール

以下のようにインストール可能なpythonのリストを確認し、

$ pyenv install --list

以下でインストールしようとした。

$ pyenv install 3.9.1

BUILD FAILEDのエラーが確認された。
以下のようにいくつかのライブラリを追加。

$ sudo apt install -y build-essential libbz2-dev libdb-dev \
  libreadline-dev libffi-dev libgdbm-dev liblzma-dev \
  libncursesw5-dev libsqlite3-dev libssl-dev \
  zlib1g-dev uuid-dev tk-dev

上記追加後、再実行

$ pyenv install 3.9.1

解決した。

###pytorch実行時のエラー

以下で新しい環境newenvの作成

$ pyenv virtualenv 3.9.1 newenv

以下でnewenvの有効化

$ pyenv local newenv

以下のようにtorchとtorchvisionのインストールを行なった

$ pip install torch==1.7.1
$ pip install torchvision=0.8.2

インストール後、GPUを使った学習を行おうとしたところ、以下のようなエラーメッセージが出現した。

GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75.

以下のようにいくつかのライブラリを追加し解決した。
追加するライブラリについてはpytorchの公式サイトで環境を選択し、取得する

$ pip uninstall torch
$ pip uninstall torchvision
$ pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

ちなみに上記をrequirements.txtに加える場合は以下の通り

requirements.txt

-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cu110
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.8.2+cu110
-f https://download.pytorch.org/whl/torch_stable.html
torchaudio===0.7.2