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

    技术2022-07-12  67

    原先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

     

     

    Processed: 0.022, SQL: 9