转载:https://blog.csdn.net/u014670893/article/details/82467384
ubuntu18.04 安装tensorflow-gpu
目录
主要参考:https://medium.com/@taylordenouden/installing-tensorflow-gpu-on-ubuntu-18-04-89a142325138
驱动
https://blog.csdn.net/tonydz0523/article/details/83109343
安装完成后最好重启下。
检查驱动是否安装成功
CUDA Toolkit 9.0
目前tensorflow只支持CUDA Toolkit 9.0。
下载地址:
目前没有支持ubuntu18.04的CUDA Toolkit 9.0。选择17.10的版本,安装base installer一般够用了。
-
sudo chmod +x cuda_9.
0.176_384.81_linux.run
-
./cuda_9.
0.176_384.81_linux.run --
override
installing with an unsupported configuration?时选择yes
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?为了稳定最好选择no
样例可以不安装。
CUDNN 7.0
地址:https://developer.nvidia.com/rdp/cudnn-archive
-
# 解压
-
tar -zxvf cudnn-
9.0-linux-x64-v7.tgz
-
-
# 复制相应文件
-
sudo cp -P cuda/lib64/libcudnn*
/usr/local
/cuda-9.0/lib64/
-
sudo cp cuda/
include/cudnn.h /usr/local/cuda-
9.0/
include/
-
-
# 所有用户可读
-
sudo chmod a+r /usr/local/cuda-
9.0/
include/cudnn.h /usr/local/cuda/lib64/libcudnn*
libcupti
sudo apt-get install libcupti-dev
配置
在~/.bashrc中加入
-
export PATH=
/usr/local
/cuda-9.0/bin${
PATH:+
:${PATH}}
-
export LD_LIBRARY_PATH=
/usr/local
/cuda/lib64
:${
LD_LIBRARY_PATH:+
:${LD_LIBRARY_PATH}}
安装tensorflow-gpu
根据需要进行修改
pip install --upgrade tensorflow-gpu
测试
成功!
目录