gpu ubuntu16

ubuntu16.04 下安装

sudo pip3 install --upgrade https://download.tensorflow.google.cn/linux/gpu/tensorflow_gpu-1.6.0-cp35-cp35m-linux_x86_64.whl

• 装显卡驱动

https://baijiahao.baidu.com/s?id=1594530812869437291&wfr=spider&for=pc

ubuntu-drivers devices

sudo apt install nvidia-384

• cuda9.0

https://developer.nvidia.com/cuda-toolkit-archive

扫描二维码关注公众号，回复： 4709655 查看本文章

环境 ：

export PATH=/usr/local/cuda-9.0/bin:$PATH export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64" export CUDA_HOME=/usr/local/cuda

• cuDNN7.1

https://developer.nvidia.com/cudnn

tar xvzf cudnn-9.0-linux-x64-v7.1.tgz sudo cp -P cuda/include/cudnn.h /usr/local/cuda/includecudnn-9.0-linux-x64-v5.1-ga sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

参考：

https://blog.csdn.net/Elliott_Yoho/article/details/78438358

https://www.cnblogs.com/Micang/p/8527483.html