当前环境：，显卡为GeForceRTX3080Ti

1、下载显卡驱动

驱动网站：

根据自己的显卡型号和系统，其中在系统中，会存在Linux64-bit和Linuxaarch64两个选项；

Linux64-bit：是一个较为宽泛的概念，涵盖了所有64位的Linux系统，可运行在多种64位硬件架构上，如常见的x86_64架构，多应用于Intel和AMD的桌面及服务器处理器。

Linuxaarch64：专门用于AArch64架构，即ARMv8及后续版本的64位ARM架构，常见于移动设备、嵌入式系统及一些低功耗服务器，如苹果的M系列芯片、树莓派等。

驱动信息

下载驱动

2、安装显卡驱动

先进入到驱动所在的文件夹，执行以下命令，让文件变为可执行文件

sudochomd+xNVIDIA-Linux-x86_64-570.144.run

安装，执行以下命令

sudo./NVIDIA-Linux-x86_64-570.144.run

安装

因为我们要安装CUDA所以选择NVIDIAProprietary，表示专有驱动版本。另一个表示开源版本。

执行下面命令

nvidia-smi

nvidia-smi

3、安装CUDA

下载cuda：

根据自己的信息选择即可，我的是得系统

cuda选择

安装cuda，从终端进入存放刚才run文件的文件夹内，然后执行下面命令

sudochmod+xcuda_12.9.0_575.51.03_/cuda_12.9.0_575.51.03_

弹出这个窗口后，输入accept，回车，进入下图

直接默认即可，按下，选中install，进行安装。

安装成功后，输入命令

nvcc--version

如上图所示，即表示安装成功，我这里之前安装过12.8得cuda，我暂时就没更新。

4、安装pytorch

pytorch安装地址：

pytorch

选择完版本后，按照命令执行，安装即可。

pip3installtorchtorchvisiontorchaudio--index-url

但往往很慢很慢..

加速

如果本地有下载快的方式(大家都懂得，小梯子)，则可按照下面的方式尝试安装

打开网址

可以查看到如图所示的，因为我的python版本为3.10，所以我选择的是箭头所指的包，大家可根据自己的环境进行选择。

同理选择torchaudio、torchvision、nvidia-cublas-cu12，选择如下图

torchaudio

torchvision

nvidia-cublas-cu12

nvidia-cudnn-cu12

nvidia-cufft-cu12

执行下面命令，进行安装

安装nvidia-cudnn-cu12pipinstallnvidia_cudnn__2_27_x86_64.whl安装+cu128-cp310-cp310-manylinux_2_28_x86_64.whl安装+cu128-cp310-cp310-manylinux_2_28_x86_64.whl

其他还会存在一些比较小的包的下载，可以直接走网络下载，都比较小。

当然如果有比较大的，也可以这样去下载

因为我们是得。所以在
，直接进行下载。

5、测试Pytorch

打开终端，输入python或者python3，进入python环境，输入下面的命令，输出结果为True，则表示安装成功

importtorchprint(_available())//True

安装成功

大功告成！！！