ubuntu 22.04配置cuda和cudnn

  • cuda:12.1

    wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
    sudo sh cuda_12.1.1_530.30.02_linux.run
    

    Abort/Continue选择Continue,不要勾选自带的driver
    配置环境变量。~/.bashrc

    export PATH=/usr/local/cuda-12.1/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH
    

    安装补充包,动态加载和卸载 NVIDIA 内核模块

    sudo apt install nvidia-modprobe
    
  • cudnn

    sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb # 安装
    sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-*-keyring.gpg /usr/share/keyrings/ # 导入GPG密钥
    sudo apt update # 更新软件包列表
    sudo apt install libcudnn8 libcudnn8-dev # 安装cuDNN运行时和开发库
    sudo ldconfig -v # 刷新ldconfig 缓存
    cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 检查cuDNN版本
    ls /usr/lib/x86_64-linux-gnu/libcudnn*	# 检查库文件
    sudo ldconfig -p | grep cudnn	# 测试动态链接
    

你可能感兴趣的:(ubuntu,linux)