AWS Cloud9にTensorFlow(GPU版)をインストール
#必要なリポジトリを取得 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub sudo apt-get update wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb sudo apt-get update #NVIDIA driverインストール sudo apt-get install --no-install-recommends nvidia-driver-410 →ここでreboot #インストール確認 nvidia-smi #TensorFlow(GPU版)をインストール pip3 install --upgrade tensorflow-gpu
これだと
import tensorflow as tf
で
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
が大量に出力される。これはtensorflowとnumpyのversionがあっていないため。tensorflow 1.14.1はnumpy 1.16.4を使うので、それにversion downする。
pip3 install numpy==1.16.4