十念 发表于 2025-3-21 23:50:53

Ubuntu22.04安装NVIDIA驱动+CUDA+CUDNN+Anaconda+PyTorch+Pycharm

前言

换新的工作呆板,需要重新设置环境,记录以下安装过程的坑
一、安装NVIDIA驱动

1.查看自己电脑型号

   lspci | grep -i nvidia
https://i-blog.csdnimg.cn/direct/51de3f5594bf414c87891c17776085da.png
 进入下方网址查询自己的显卡
PCI Devices
https://i-blog.csdnimg.cn/direct/ca2a747c5cea464db6183e0a619203af.png
2.进入NVIDIA官网下载驱动

 下载 NVIDIA 官方驱动 | NVIDIA
https://i-blog.csdnimg.cn/direct/6306b6fcc2e74a4f817ddb581d3659c0.png
 选择下载之后得到一个后缀名为run的文件
https://i-blog.csdnimg.cn/direct/5f6b69e1a1364ceab97deacc5edb71c0.png
3.准备工作下载依赖

 1)禁用BIOS中的安全启动模式secure boot,这是因为此方法使用第三方源安装显卡驱动,不禁止secure boot会导致安装的驱动不能使用,禁用也不会有多大安全隐患。不禁用背面会导致安装出现error并且fail。具体做法是:

开机进入BIOS,然后选择安全模式,禁止。
2)移除所有以前安装过的nvidia包(假如需要的话)

sudo apt-get purge nvidia* 3)禁用nouveau;创建文件,方式选择下面方法之一,一般人很少用vim(vim是a大概i开启编辑,esc退出,输入:wq生存退出),风俗用自带的gedit命令直接打开相当于记事本那样编辑:

sudo nano /etc/modprobe.d/blacklist.conf sudo gedit /etc/modprobe.d/blacklist.conf sudo vim /etc/modprobe.d/blacklist.conf 插入代码到blacklist.conf文本最下面,如下
blacklist nouveau
options nouveau modeset=0 https://i-blog.csdnimg.cn/direct/24414a0802fd4b388d31473d06c05440.png
执行命令禁止nouveau,并重启见效
sudo update-initramfs -u
sudo reboot
重启后,打开终端检查nouveau是否被禁用;
输入下面代码验证,若执行完该句,没有任何输出,则nouveau被乐成禁用。
lsmod | grep nouveau
4) 在安装驱动之前需要安装gcc和相应的依赖:

sudo apt-get install build-essential
sudo apt-get install cmake
 查看gcc和cmake有没有安装乐成:
gcc --version
cmake --version
5) 没有lightdm需要安装一下:

sudo apt-get install lightdm
6) 附:上述安装假如一直安装不下来,于是换镜像源,更换前最好先备份:

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak sudo nano /etc/apt/sources.list
将原有的源都加#注释,在最后换成自己想要的源,下面的是阿里云
deb http://mirrors.aliyun.com/ubuntu/ focal main restricted
deb http://mirrors.aliyun.com/ubuntu/ focal-updates main restricted
deb http://mirrors.aliyun.com/ubuntu/ focal universe
deb http://mirrors.aliyun.com/ubuntu/ focal-updates universe
deb http://mirrors.aliyun.com/ubuntu/ focal multiverse
deb http://mirrors.aliyun.com/ubuntu/ focal-updates multiverse
deb http://mirrors.aliyun.com/ubuntu/ focal-backports main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ focal-security main restricted
https://i-blog.csdnimg.cn/direct/5d590c3f9a3844c8af892e5aabb6c004.png
https://i-blog.csdnimg.cn/direct/bb12cd4539464c5e9784ff0375a1aa56.png
更换源后需要更新
sudo apt-get update
sudo apt-get upgrade 4.统统停当,开始安装

打开NVIDIA-Linux-x86_64-570.86.16.run文件地点目录,右键在终端打开大概cd到NVIDIA-Linux-x86_64-570.86.16.run文件地点的文件夹
https://i-blog.csdnimg.cn/direct/5f6b69e1a1364ceab97deacc5edb71c0.png
 运行如下命令,修改驱动文件的权限并执行安装
sudo chmod +x NVIDIA-Linux-x86_64-570.86.16.run sudo ./NVIDIA-Linux-x86_64-570.86.16.run 会载入一段时间,然后表现,安装过程中会进入一个交互式界面,按键盘的左右箭头选择相应选项,按回车键确认:
https://i-blog.csdnimg.cn/direct/300709df45314433a534c6a9d418b581.png
回车下一步,选择continue installation
https://i-blog.csdnimg.cn/direct/bb76a83e5d024a70816f2b684f9b86b3.png
接下来,系统会询问是否需要内核签名。此时,选择 Sign the kernel(签名内核)回车继续
https://i-blog.csdnimg.cn/direct/f7996c0bb22d4c8785a6fe615b739d5a.png
系统将提示你创建新的密钥对,选择 Create new key pair(创建新密钥对)这就会在/usr/share/nvidia/下天生一个nvidia*.der文件和nvidia*.key文件,前一个是私钥,后一个是公钥。
https://i-blog.csdnimg.cn/direct/3136c4702a5a4ebeb45d0b46d38f944e.png
完成签名后,删除密钥:
https://i-blog.csdnimg.cn/direct/b560a52a2b6f441c888f8ae42c9f0aaf.png
注意:假如出现以下提示,阐明由于系统的 BIOS 启用了 Secure Boot,这可能会导致内核加载失败。此时,您需要进入 BIOS 设置中禁用 Secure Boot
https://i-blog.csdnimg.cn/direct/308246d733e84f6a9ef019c7f7903ff8.png
安装完成后,系统会询问是否需要安装 nvidia-x-config,该工具会提供一个图形界面的显卡管理工具。发起选择 Yes 举行安装。
https://i-blog.csdnimg.cn/direct/164ff29ba0a946f4a042665afcd4324f.png
安装完成后,重启系统:
sudo reboot
 重启后,使用以下命令检查显卡状态,出现下图表示安装乐成:
nvidia-smi
https://i-blog.csdnimg.cn/direct/8f49dfba28b04e49ba01602ec287126f.png
5.倒霉用安装包安装,特别慢

按照顺序执行下面代码
sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update 指定版本安装用下面这个
sudo apt install nvidia-driver-570 自动读取设置安装用这个
sudo ubuntu-drivers autoinstall 二、Cuda安装

1.选择适合自己的版本号,下面官网中挑选

CUDA Toolkit 12.4 Update 1 Downloads | NVIDIA 开发者
https://i-blog.csdnimg.cn/direct/c967aaac3128449ea4bf9beeaabbae32.png
2.终端使用如上图所示指令举行下载:

wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run

sudo sh cuda_12.4.1_550.54.15_linux.run 安装时,仍会提示目的CUDA toolkit对应版本的显卡驱动未被发现,推荐我们安装其版本的显卡驱动,这时我们键入continue,再键入accept同意用户协议,然后将光标移动到driver处,键入回车,以取消对驱动的安装:其他的保持不动,选择“Install”,稍等几分钟就装好了
https://i-blog.csdnimg.cn/direct/edba3413c3e04a618a8e45b1d7a25ab5.png https://i-blog.csdnimg.cn/direct/6918901e919f410bb76d870ff6ed080e.png
出现此画面,表示安装乐成
https://i-blog.csdnimg.cn/direct/9a3bf17cd0ae4f1da4020e7385a0f86b.png
3.设置环境变量

sudo gedit ~/.bashrc 在尾部加入以下内容(路径不肯定和我一样,看版本号)
export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 更新环境变量
source ~/.bashrc https://i-blog.csdnimg.cn/direct/bb53d473cd2c4259bc8ca75d907c472b.png
4.查看CUDA版本信息

nvcc -V https://i-blog.csdnimg.cn/direct/6d2d0133dcd44e37ab088480b9c02440.png
5.测试CUDA(非必要) 

打开下载位置,随便运行一个
https://i-blog.csdnimg.cn/direct/3a6dcbcce5bb4b268f683bcd2f34ad36.png 比方bandwidthTest,看到下图阐明安装乐成https://i-blog.csdnimg.cn/direct/c3fcd6fa38ee48fc82e9463e3822bfcf.png
三、cuDNN安装

1.选择适合自己的的cuda版本和系统举行下载

cuDNN Archive | NVIDIA Developer
https://i-blog.csdnimg.cn/direct/bf70ef85ee9b41058eb2a88a900fc956.png
完成后,出现cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb
https://i-blog.csdnimg.cn/direct/5986a304666a4a9e9e5bca039e0e3360.png
2.用终端打开,执行命令等待完成

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb 四、安装Anaconda

1.下载完成之后,进入相应目录

https://i-blog.csdnimg.cn/direct/ba258379f24a4166bf110951168d1640.png
2.运行如下指令,举行安装

bash ./Anaconda3-2024.10-1-Linux-x86_64.sh  sh Anaconda3-2024.10-1-Linux-x86_64.sh
1、安装过程中根据提示输入enter; 
2、最后一部问是否添加环境变量,最好选择yes;
3、验证安装(重开终端窗口)(假如不可可能是环境变量没加进去)(或source环境)
conda -V https://i-blog.csdnimg.cn/direct/b9ea6305ace14f43b64f98a053c67d55.png
3.常用命令

查看当前所有的env
conda env list 创建新的env
比方这个语句就是创建一个名字叫nv的,python版本为3.9的一个虚拟环境
conda create -n nv python=3.9 https://i-blog.csdnimg.cn/direct/decf3ff0dcb24a6a9c65ec6d02481eff.png
激活环境命令
conda activate pt 退出环境命令
conda deactivate 五、安装Pytorch

1.下载位置

Previous PyTorch Versions | PyTorch
https://i-blog.csdnimg.cn/direct/5450623862c24907923cd781de8d328c.png
2.安装命令

conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 pytorch-cuda=12.4 -c pytorch -c nvidia
假如进入自己的conda虚拟环境,激活后,运行上方指令。
六、PyCharm安装及使用torch

1.下载位置

PyCharm: the Python IDE for data science and web development
下载解压之后,进入解压目录,进入bin目录,./pycharm.sh可以运行,可以在内里设置快捷方式;
在工具栏Tools,有Create Desktop Entry..,可以创建快捷方式。
2.新建工程

选择
https://i-blog.csdnimg.cn/direct/ebe99017093742668a179c376ac3986e.png
选择环境使用conda,然后Interpreter选择自己刚才创建的虚拟环境 
3.代码测试torch

import torch

print(torch.cuda.is_available())
a = torch.zeros(5, 3)
print(a)
七、参考链接

Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch_ubuntu20.04安装cuda10.2-CSDN博客
Ubuntu18.04安装Nvidia驱动【全网不坑,超全步骤】(亲测~)_ubuntu18.04安装nvidia显卡驱动-CSDN博客 Ubuntu22.04安装CUDA+CUDNN+Conda+PyTorch_ubuntu22安装cuda conda-CSDN博客
Ubuntu20.04安装CUDA+CUDNN+Conda+PyTorch_ubuntu20.04安装cuda10.2-CSDN博客
【深度学习环境设置】ubuntu 20.04+4060 Ti+CUDA 11.8+pytorch(装机、显卡驱动、CUDA、cudnn、pytorch)_4060ti cuda cudnn-CSDN博客 Ububtu20.04 CUDA安装与环境设置、CUDNN安装_ubuntu20.04查看cunda-CSDN博客
Ubuntu24版 最新安装Nvidia显卡驱动方式_ubuntu安装nvidia显卡驱动-CSDN博客
总结

新装备安装深度学习环境。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Ubuntu22.04安装NVIDIA驱动+CUDA+CUDNN+Anaconda+PyTorch+Pycharm