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

CUDA Toolkit 11.7 安装指南

原创已于 2022-08-11 09:10:18 修改 · 728 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #ubuntu #人工智能

于 2022-08-10 14:47:18 首次发布

本文提供了一步一步的指南，帮助读者完成CUDA Toolkit 11.7.0及CUDNN 8.5.0的安装过程。从选择合适的国内源开始，介绍了如何安装NVIDIA显卡驱动，以及如何根据显卡信息挑选正确的CUDA版本。此外，还详细说明了将CUDA添加到环境变量的方法，并给出了安装CUDNN的具体步骤。

部署运行你感兴趣的模型镜像

CUDA Toolkit11.7.0 + CUDNN8.5.0

换国内源
setting->about->Software Updates->Ubuntu Software->Download from->Other->自己选
参考：https://cloud.tencent.com/developer/article/1726035
安装nvidia显卡驱动
setting->about->Software Updates->Additional Drivers->自己选
注意：如果显卡驱动没选择对，会导致显示器显示出问题，无法选择合适的分辨率
安装cuda
在成功安装显卡驱动的情况下，打开terminal，使用nvidia-smi查看显卡信息，CUDA Version: 11.7
在https://developer.nvidia.com/cuda-toolkit-archive网址下找自己合适的cuda toolkit
把cuda添加到环境变量中，https://blog.youkuaiyun.com/Flying_sfeng/article/details/103343813
在terminal中使用nvcc -V查看目前使用cuda版本。
安装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/
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