1.直接到NVIDA官网下载相应的驱动，然后安装官方驱动 | NVIDIA

2.下载相应版本cuda，并安装，安装时不安装驱动

3.conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia 安装pytorch。

安装之前先部署环境：conda create  -n pytorch python=3.11

因为安装的pytorch版本是2.1.0。

注意：这里pytorch-cuda=12.1

但是：

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Tue_Jun_13_19:16:58_PDT_2023
Cuda compilation tools, release 12.2, V12.2.91
Build cuda_12.2.r12.2/compiler.32965470_0

这里显示12.2不影响。

验证：

(pytorch) wawayu@wawayu:~$ python
Python 3.11.5 (main, Sep 11 2023, 13:54:46) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.1.0
>>> torch.cuda.is_available()
True