赞
踩
一、安装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是否安装成功
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。