首页 > 系统相关 >Ubuntu+Cuda+cudnn安装

Ubuntu+Cuda+cudnn安装

时间:2024-10-04 15:35:50浏览次数:12  
标签:Cuda nvidia sudo cudnn usr Ubuntu local cuda

问题描述

实验室的Ubuntu主机重启之后显示器分辨率显示异常,双屏只亮了一个,另外一个显示无信号。
打开nvidia-smi显示找不到驱动,遂从网上搜集重新安装驱动以及cudnn的教程。此文记录此过程,方便后续查阅。

软硬件版本:Rtx3090 + Ubuntu22.04

1. 安装驱动

# remove nivida*
sudo apt-get --purge remove nvidia*

# install nvidia-driver
# 安装指定版本:
sudo apt install nvidia-driver-525
# 安装推荐版本:
sudo ubuntu-drivers autoinstall

2. 安装cuda

CUDA Toolkit 11.7 Downloads

# 下载需要的版本
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

# 设置环境变量
sudo gedit ~/.bashrc

# 添加
export PATH="/usr/local/cuda-11.7/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH"

# 激活
source ~/.bashrc

# reboot
rebbot

# then
nvidia-smi

3. 安装cudnn

cuDNN Archive Downloads

# 下载cuDNN v8.9.7 for CUDA 11.x
wget https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.7/local_installers/11.x/cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

# 解压
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

# 将cudnn.h等文件复制到 /usr/local/cuda中
cd cudnn-linux-x86_64-8.9.7.29_cuda11-archive
sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

# 测试安装是否成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

参考文章:
https://zhuanlan.zhihu.com/p/711028014
https://blog.csdn.net/sinat_16423171/article/details/140351150

标签:Cuda,nvidia,sudo,cudnn,usr,Ubuntu,local,cuda
From: https://www.cnblogs.com/eewpan/p/18446596

相关文章

  • 25赛季算法组第一阶段第二次培训(ubuntu安装与基本使用)
    25赛季算法组第一阶段第二次培训1.Ubuntu的介绍1.1.操作系统和操作系统的选择操作系统,英文名称OperatingSystem,简称OS,是计算机系统中必不可少的基础系统软件,它是应用程序运行以及用户操作必备的基础环境支撑,是计算机系统的核心。操作系统的作用是管理和控制计算机系统中的......
  • Docker配置代理访问网络ubuntu24.04
    本文将详细介绍如何根据系统代理配置,正确设置Docker的代理环境变量,使其能够通过代理服务器进行网络访问。一、查看系统代理配置首先,我们查看了系统的代理配置:以下是图片内容的文字描述:Proxy设置NetworkProxy:已开启Configuration:手动(Manual)HTTPProxyURL:12......