• Ubuntu安装任意版本的cuda和cudnn


    这篇文章记录了如何在Ubuntu22.04上安装CUDA11.7和cudnn8.6。按照步骤理应可以安装成功。如果还有问题,可以留言。

    安装显卡驱动

    打开‘软件和更新。
    image_5a9feee1.png
    点击附加驱动安装显卡驱动
    image_8dfd1201.png
    如果已经安装显卡驱动,请忽略上面的步骤。

    安装gcc

    新安装的Ubuntu22.04 没有安装gcc,需要安装gcc。在终端输入gcc -version 查看有没有gcc。
    image_e79a4973.png
    执行命令

    sudo apt install gcc
    
    • 1

    安装CUDA

    输入nvidia-smi查看支持CUDA的版本,通过下图得知,支持的最高版本是11.8.
    image_80b85e2e.png
    我在这里选中11.7的版本,主要因为pytorch目前支持的最高版本是11.7.
    CUDA下载连接:https://developer.nvidia.com/cuda-11-7-1-download-archive
    image_ae6bdba5.png
    选择对应的版本,然后在下面就可以安装命令:

    wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_515.65.01_linux.run
    sudo sh cuda_11.7.1_515.65.01_linux.run
    
    • 1
    • 2

    下载完成后,执行sudo sh cuda_11.7.1_515.65.01_linux.run
    image_79a47d10.png
    选择Continue

    image_b0049d3b.png
    输入accept

    image_b4aeb087.png点击空格,取消安装驱动,然后选择Install

    image_d9c0972d.png
    看到上面的消息,就说明安装完成了。
    配置环境变量

    输入gedit ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。

    export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    
    • 1
    • 2

    image_94c3cdd6.png
    然后,点击保存后关闭文档,然后执行命令source ~/.bashrc
    至此cuda安装完成,输入nvcc -V命令查看cuda信息。
    image_ae141f34.png

    安装cudnn

    下载链接:https://developer.nvidia.com/rdp/cudnn-download
    image_ada1979b.png然后选择红框的链接下载。等待下载完成。
    更改usr/local/cuda/文件夹下面的include和lib64文件夹的权限。
    常用的修改权限的指令

    sudo chmod 600 ××× (只有所有者有读和写的权限)
    sudo chmod 644 ××× (所有者有读和写的权限,组用户只有读的权限)
    sudo chmod 700 ××× (只有所有者有读和写以及执行的权限)
    sudo chmod 666 ××× (每个人都有读和写的权限)
    sudo chmod 777 ××× (每个人都有读和写以及执行的权限)
    
    • 1
    • 2
    • 3
    • 4
    • 5

    执行修改文件夹权限的命令:

    cd /usr/local/cuda
    sudo chmod 666 include
    
    • 1
    • 2

    然后解压cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz文件。等待解压完成后,在下载文件夹中右键打开终端。执行如下命令:

    sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
    sudo cp cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
    sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
    
    • 1
    • 2
    • 3

    image_53bc1da8.png验证是否安装成功,执行命令:

    sudo cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
    
    • 1

    image_0fea9e45.png安装成功

  • 相关阅读:
    1.图形学-矩形中的线段的裁剪Cohen-Sutherland算法(附带源码)
    人工智能之地形导航系统
    【活动总结】0730-COC深圳社区AI●CMeetup第4期——畅谈AI+智能制造与机器人的现状与未来
    【开源】基于SpringBoot的海南旅游景点推荐系统的设计和实现
    【Java SE】多态的详解
    吴恩达-机器学习-k-means聚类算法
    线程安全使用 HashMap 的四种技巧
    第二部分—C语言提高篇_12. 动/精态库的封装和使用
    软考入门级了解(时间,费用,流程),无广告成分
    vue的ref使用技巧☞重置表单
  • 原文地址:https://blog.csdn.net/u014297502/article/details/127827648