Ubuntu22.04.2 LTS 安装nvidia显卡驱动及配置pytorch

首先安装显卡驱动

首先，查看我们可以使用的驱动版本，使用如下指令。可以看到打印出来的可以安装的版本号

sudo ubuntu-drivers devices

之后安装一个较新版本的驱动，使用如下指令。

$ sudo apt install nvidia-driver-470

安装完成后，直接nvidia-smi可能并不显示，提示要运行驱动，这时候只要重启一下，再输入nvidia-smi即可看到显卡信息，完成安装。

安装cuda

参考我的另一篇文章中，有介绍cuda的安装。

MindSpore服务器环境安装___glibc[version='>=2.17,<3.0.a0']_失落的熊熊的博客-CSDN博客

安装anaconda3

可以从官网进行下载，或者其它的清华源等。下面是官网的下载地址。

Free Download | Anacondax

下载完最新的安装包之后，添加环境变量。使用ubuntu自带的gedit，比vi，vim要友好很多。

gedit ~/.bashrc

 在文本中加入如下一句

export PATH="/home/ssj/anaconda3/bin:$PATH"

然后source一下，更新

source ~/.bashrc

如果顺利，直接就出来base的环境了。可以查看conda版本。

conda --version
conda -V

安装pytorch

既然安装了conda，那就先创建一个conda环境吧，可以参考我另一篇文章

CONDA常用指令及测试pytorch是否安装成功_conda测试pytorch_失落的熊熊的博客-CSDN博客

激活环境，然后使用conda命令安装pytorch，大家可以参考pytorch的官网，看最新的版本如何安装。

PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

如果是cpu版本，可以查看我的另一篇文章

PyTorch AttributeError: module ‘torch._C‘ has no attribute ‘_cuda_setDevice‘_失落的熊熊的博客-CSDN博客之后大家就可以用自己的电脑跑深度学习了.....

emmmm，谁会用自己的电脑来跑实验呢？