python 深度学习虚拟环境conda安装cuda11.7和cudnn和pytorch

一、安装Anaconda

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

二、配置虚拟环境

conda create -n pcdet python=3.9 # 创建python3.9环境

conda activate pcdet # 激活环境

pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117 #安装pytorch

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

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

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

但此时环境中仍没有cuda如图

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

conda install -c nvidia cudatoolkit

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

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

conda install cudnn

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

金钥匙