ubuntu16.04+cuda10.1+cudnn7.6.5+tensorflow2.2.0

原先cuda10.0，安装10.1。

～/.bashrc。

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

 

修改cuda软链接。

    rm -rf /usr/local/cuda   #删除之前创建的软链接
    sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda
    nvcc --version    #查看当前 cuda 版本

下载cudnn7.6.5，runtime和developer，放了include和lib64文件。

卸载

pip uninstall tensorflow-gpu==1.14

安装

pip install -i https://pypi.doubanio.com/simple tensorflow-gpu==2.2.0