マガジンのカバー画像

Python関連

6
運営しているクリエイター

#tensorflow

Tensorflow 2.x & PyTorch 1.6でGPUを使う【Ubuntu 20.04】

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

もっとみる