- 投稿日:2020-07-25T13:26:04+09:00
NVIDIA CUDA toolkit on Ubuntu20.04
システム構成
- PC:DELL XPS15 2017(GeForce GTX 1050)
- OS:Ubuntu Linux 20.04 LTS Desktop 日本語版
CUDA Toolkitインストール
このインストール方法は Anaconda の Python を利用しないと使えないかもしれないです。
conda install でAnaconda で用意されている CUDA と cuDNN をインストールしているように見えます。
- Ubuntu のログインメニューから「ソフトウェアのアップデート」を開く
「追加のドライバー」タブで「NVIDIA driver metapackageをnvidia-driver-440から使用します(プロプラエタリ)」を選択して、「変更の適用」ボタンを押下
CUDA Toolkit のインストール
$ sudo apt-get install $ sudo apt install nvidia-cuda-toolkitAnacondaのインストール
sh ./Anaconda3-2020.02-Linux-x86_64.shTensorflow、Pytorch、CuPyのインストールと動作確認
Tensorflow1系と2系は同居できないので、両方を使いたい場合は仮想環境を作る必要があります。
あと、よくわかりませんが、python3.8 に tensorflow1.14 はインストールできないようです。
- Tensorflow1系
$ conda create --name=tf1 python=3.7 jupyter $ conda activate tf1 $ conda install tensorflow-gpu==1.14.0 $ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch $ pip install cupy-cuda101 chainer chainercv
- Tensorflow2系
$ conda create --name=tf2 jupyter $ conda activate tf2 $ conda install tensorflow-gpu $ conda install pytorch torchvision cudatoolkit=10.1 -c pytorch $ pip install cupy-cuda101 chainer chainercvGPUの動作確認
- Tensorflow
python -c 'from tensorflow.python.client import device_lib; device_lib.list_local_devices()'
- Pytorch
python -c 'import torch; print(torch.cuda.is_available())'
- CuPy
$ python -c 'import chainer; chainer.print_runtime_info()'その他
Ubuntu がGPUを認識しているかの確認
$ sudo ubuntu-drivers devicesGPUドライバが正しくインストールされているかの確認
$ nvidia-smi