- 運営しているクリエイター
#gpu
Tensorflow 2.x & PyTorch 1.6でGPUを使う【Ubuntu 20.04】
更新2020.09.05 pipenvでPyTorchのインストール方法を変更した
この記事が目指す環境・Ubuntu 20.04 LTS
・pyenv (Pythonのバージョン管理ツール)
・Python 3.x (3.6以上推奨)
・pipenv (仮想環境の作成等)
・Tensorflow 2.x (GPU対応)
・PyTorch 1.6.x (GPU対応)
・CUDA 10.1
・Cu