Ubuntu 18.04 NVIDIAドライバをインストールし、CUDAを配置し、Tensorflowフロー

2466 ワード

参照リンク
http://homeway.me/2017/08/08/setup-gpu-for-tensorflow/
 
ロートでインストール中
文章の部分はバッグをインストールして自分の状況によってダウンロードする必要があります.伝言や私信が必要です.あなたに送ることができます.愛のあるゲストは食事代を与えることができます.
 
1.グラフィックカードの設定を確認する
lspci | grep -i vga
 
2.対応するグラフィックカードドライバをダウンロードする
https://www.nvidia.com/Download/index.aspx?lang=en-us
Nvidiaドライバに依存環境を構築する 
sudo apt-get install dkms build-essential linux-headers-generic


sudo mkdir /etc/modprobe.d/

sudo touch /etc/modprobe.d/blacklist-nouveau.conf


cat >> /etc/modprobe.d/blacklist-nouveau.conf << EOF

blacklist nouveau

blacklist lbm-nouveau

options nouveau modeset=0

alias nouveau off

alias lbm-nouveau off

EOF
sudo vi blacklist-nouveau.com nfが読めないなら
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

sudo update-initramfs -u


sudo reboot
NVIDIA-Linux-XXXXは対応するグラフィックカードのドライババージョンで、修正が必要です.
sudo chmod u+x NVIDIA-Linux-x86_64-390.77.run

sudo ./NVIDIA-Linux-x86_64-390.77.run

sudo reboot
3.クダを取り付ける
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub

sudo apt-get update

sudo apt-get install cuda -y


CUDA_HOME=/usr/local/cuda-10.0

ln -vfs $CUDA_HOME /usr/local/cuda

cat >> ~/.profile << EOF

export CUDA_HOME=${CUDA_HOME}

export LD_LIBRARY_PATH=${CUDA_HOME}/lib64

PATH=${CUDA_HOME}/bin:${PATH}

EOF

source ~/.profile
ここの経験は私達に再起動が必要だと教えてくれます.
 
テストCUDA
λ$cd$CUDA_HOME/samples/1_Utilities/deviceQuery
⑧make
⑧$/deviceQuery
 
4.CUDA NNの設置
tar xzvf cudnn-8.0-linux-x64-v5.1.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
5.Pythonの設置と依存環境
sudo apt-get install python-dev python3-dev python-numpy \

 python3-numpy python-six python3-six build-essential python-pip python3-pip


sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb

sudo dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb

sudo apt-get update


dpkg -i  nccl-repo-ubuntu1804-2.3.5-ga-cuda10.0_1-1_amd64.deb

apt-get update
6.tenssor flowを取り付ける
pip 3 install tenssorflow-1.10 rc-c 36-cp 36 m-linux_x 86_64.whl
numpyを新たに取り付ける必要があります.
pip 3 install-U numpy