一、配置conda以及cuda和cudnn和pytorch

Ubuntu 20.04安装Anaconda3及简单使用_ubuntu anaconda3使用-CSDN博客

首先是根据上面的内容进行conda的安装，方便之后创建虚拟环境，在虚拟环境中安装pytorch来方便后续使用。

ubutnu20.04配置深度学习环境-CSDN博客

安装的一个主体思路，是首先安装好conda后，在conda中创建一个环境来进行pytorch的安装

然后依次conda create -n 环境名称 python=3.12

source activate 环境名称 来激活环境

然后安装cuda，首先是拷贝过来cuda11.3的run文件，然后sudo chmod +x 文件名称

sudo ./cuda的文件名称

然后点continue，输入accept回车，空格把显卡驱动取消掉，到install里面回车

打开终端输入gedit ~/.bashrc

export PATH=/usr/local/cuda-11.3/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64

export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH

然后刷新一遍环境变量source ~/.bashrc

然后需要安装pytorch

在conda activate +虚拟环境中运行pip install 加对应的torch轮子进行pytorch的安装

然后依次调用python，import torch ，import torchvision ,print(torch.__version__),print(torch.version.cuda),print(torch.backends.cudnn.version())来验证是否安装成功。