最近又重新配置了深度学习环境，重新记录一下过程

超级感谢 博主指路原教程：整个系列走下来很顺畅！！https://blog.csdn.net/weixin_58283091/article/details/135589017?spm=1001.2014.3001.5501

ubuntu系统为Ubuntu 22.04.4 LTS

1 安装好显卡驱动并查看显卡驱动版本

nvidia-smi

2 安装cuda

查看cuda和显卡驱动对应版本（对照官方文档）

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

选择满足版本条件的cuda下载

 https://developer.nvidia.com/cuda-toolkit-archive

continue->accept->取消勾选驱动（已安装） 

跳回原页面等待

显示这个就成功了 继续修改配置文件

sudo gedit /etc/profile

在文档末尾增加： 这里路径替换为上图框选的

export PATH=$PATH:/usr/local/cuda-12.2/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.2/lib64

（变色）成功！

3安装cuDNN

https://developer.nvidia.com/rdp/cudnn-archive

下好后提取解压缩。

tar -xvf cudnn-linux-x86_64-8.9.6.50_cuda12-archive.tar.xz 

然后，使用下面指令复制cuDNN文件夹下（以自己实际安装路径为准）。tab补全自动填下载的版本

sudo cp cudnn-linux-x86_64-8.9.6.50_cuda12-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -p cudnn-linux-x86_64-8.9.6.50_cuda12-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

4 安装Pytorch（建议参考原文档）

这里我是直接根据项目需求安装：项目提供了requrement文件。

pip install -r requirements/pyt211_cu121.txt

报错后单独重新安装

conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=12.1 -c pytorch -c nvidia

检验结果如下：

python
Python 3.9.24 (main, Oct 21 2025, 20:11:42) 
[GCC 11.2.0] :: Anaconda, Inc. on linux                                         
Type "help", "copyright", "credits" or "license" for more information.          
>>> import torch
>>> print(torch.version.cuda)
12.1                                                                            
>>> print(torch.backends.cudnn.version())
8902                                                                            
>>> print(torch.__version__)
2.1.1                                                                           
>>> exit()