一、安装Anaconda

参考文章：Ubuntu20.04安装anaconda_ZCONG呀的博客-CSDN博客

二、配置虚拟环境

conda create -n pcdet python=3.9  # 创建python3.9环境
conda activate pcdet # 激活环境
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia

推荐使用NVIDIA官方提供的一键安装torch+torchvision+cudatoolkit，这样可以减少安装cuda的麻烦 ，详细解释可参考：一文讲清楚CUDA、CUDA toolkit、CUDNN、NVCC关系_健0000的博客-CSDN博客

注意！！关于版本问题，博主先安装的是cu116的版本对应的torch，但之后在安装cudnn时发现只能安装CUDA11.7，为了防止pytorch与cuda版本不对应，于是就卸载重装

安装完后应显示如图则表示安装成功

查看此时环境中所有包

 

 可以看到已有CUDA包和torch，表示安装成功。

但有的人可能环境中仍没有cuda如图

 于是在anaconda官网Cudatoolkit :: Anaconda.org下载

conda install -c nvidia cudatoolkit

注意！！网上安装cudatoolkit有很多方法，但大多已不使用，最便捷的方式是通过官网指令下载

再下载对应的cudnn，一定要先下载cudatoolkit再下cudnn，否则在下载cudnn时会同时下载cudatoolkit，导致版本问题！

conda install cudnn

三、验证pytorch、cuda、cudnn是否安装成功