如何在ubuntu16.04安装TensorflowGPU

xiaoxiao2021-02-28  15

我这里配置的是tensorflow1.6.0 + cuda9.0 + cudann 7.1.1

1.更改驱动

在“软件和更新”里找到 附加驱动把驱动换成NIVIDA BINARY 我的是384.111

2.安装cuda

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

这里是cuda9.0的下载地址,英伟达有对应的安装指令 输入即可

3.安装cudann

登录后在这里下载 对应打这个

cuDNN v7.1.1 Library for Linux

用tar指令解压

然后

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

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

最后sudo gedit  ~/.bashrc

加上

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"export CUDA_HOME=/usr/local/cuda

export PATH=/usr/local/cuda/bin:$PATH

4.装好tensorflowGPU环境 就好了

视频教程:

https://www.youtube.com/watch?v=fY0n0WD3UXs

转载请注明原文地址: https://www.6miu.com/read-2629828.html

最新回复(0)