環境
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