Ubuntu20.04配置CUDA+CUDNN深度学习环境

CUDA Toolkit11.7.0 + CUDNN8.5.0

  1. 换国内源
    setting->about->Software Updates->Ubuntu Software->Download from->Other->自己选
    参考:https://cloud.tencent.com/developer/article/1726035

  2. 安装nvidia显卡驱动
    setting->about->Software Updates->Additional Drivers->自己选
    注意:如果显卡驱动没选择对,会导致显示器显示出问题,无法选择合适的分辨率

  3. 安装cuda
    在成功安装显卡驱动的情况下,打开terminal,使用nvidia-smi查看显卡信息,CUDA Version: 11.7
    在https://developer.nvidia.com/cuda-toolkit-archive网址下找自己合适的cuda toolkit
    把cuda添加到环境变量中,https://blog.csdn.net/Flying_sfeng/article/details/103343813
    在terminal中使用nvcc -V查看目前使用cuda版本。

  4. 安装cudnn
    注册登陆下载合适的cudnn:https://developer.nvidia.cn/zh-cn/cudnn
    安装教程:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#download
    在官网的测试指南通不过:https://www.codeleading.com/article/29255558606/

  5. cuda和cudnn官方安装指南与测试
    https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#download
    https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

CUDA Toolkit 11.6.0

猜你喜欢

转载自blog.csdn.net/BubbleCodes/article/details/126261901