当前位置:   article > 正文

虚拟环境conda安装cuda11.7和cudnn和pytorch_pytorch cuda11.7

pytorch cuda11.7

一、安装Anaconda

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

二、配置虚拟环境

  1. conda create -n pcdet python=3.9 # 创建python3.9环境
  2. conda activate pcdet # 激活环境
  3. 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是否安装成功

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/173384
推荐阅读
相关标签
  

闽ICP备14008679号