Python: GPU を使う主要なパッケージで CUDA が有効か確かめる方法について - CUBE SUGAR CONTAINER

CUDA を利用する Python のパッケージは、多くの場合それ自体のバージョンだけでなく対応している CUDA Toolkit や cuDNN のバージョンまで気にする必要がある。 なんだか環境やコンテナイメージを作るたびに確認する方法や互換性について調べている気がするので、以下のパッケージについてまとめておく。 PyTorch TensorFlow (2.x) CuPy CuDF 使った環境は次のとおり。 $ lsb_release -a No LSB modules are available. Distributor ID: Ubuntu Description: Ubuntu 18.04.5 LTS Release: 18.04 Codename: bionic $ uname -r 4.15.0-124-generic $ python -V Python 3.8.6 $ p