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
环境 :
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