在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

这篇具有很好参考价值的文章主要介绍了在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

本人最近接触深度学习,想在服务器上配置深度学习的环境,看了很多资料后总结出来了对于新手比较友好的配置流程,创建了一个关于深度学习环境配置的专栏,包括从anaconda到cuda到pytorch的一系列操作,专栏中的另外两篇文章如下,如果有不对的地方欢迎大家批评指正!

Anaconda保姆级安装配置教程(新手必看)

如果你还不是很清楚CUDA,CUDATookit,cuDNN,Pytorch分别在深度学习环境中的定位与关系,推荐看一下我这篇文章
用人话讲解深度学习中CUDA,cudatookit,cudnn和pytorch的关系

另外,如果你还不是特别理解虚拟环境的作用,建议看一下这位大佬的视频安装不算完事,只有理解了虚拟环境才算真正掌握 Python 环境_哔哩哔哩_bilibili

首先,我们要明确,我们是要在虚拟环境中安装cuda和cuDNN!!!只需要在虚拟环境中安装就可以了。

下面的操作默认你安装好了python

一、conda创建并激活虚拟环境

前提:确定你安装好了anaconda并配置好了环境变量,如果没有,网上有很多详细的配置教程,请自行学习

在cmd命令提示符中输入conda命令查看anaconda

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

如果显示和上图相同,那么可以继续向下看

1.进入anaconda的base环境

方法1——在cmd命令提示符中输入如下命令

activate

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

方法2——直接在搜索栏里搜索anaconda prompt并打开即可

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

方法3——如果你是在linux操作系统下,在你的/anaconda/bin/下打开终端,即可直接进入anaconda的环境

以上三种方法出现(base)就意味着你已经进入了anaconda的基础环境

2.conda创建、进入和退出虚拟环境

这三个操作可以各用一行命令来完成

#创建一个虚拟环境
conda create -n [your_env_name(你的虚拟环境的名字)] python==[X.X](2.5、3.8等)
#eg:conda create -n nnunet_env python=3.8

#进入虚拟环境
conda activate [你的虚拟环境名]

#退出虚拟环境
conda deactivate

创建好的虚拟环境文件夹可以在anaconda文件夹中的envs文件夹里找到

后续的环境配置操作均要在激活虚拟环境的情况下完成!!!!!

二、查看CUDA版本

系统的CUDA版本,决定了系统最高可以支持什么版本的cudatoolkit,它是向下兼容

我们可以通过nvidia-smi命令查看cuda版本号
在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

比如我的CUDA Version=11.7,那么它就可以支持 ≤ 11.7版本的CUDATookit

现在你已经知道了自己系统的cuda版本,接下来我会以自己利用清华镜像源来配置cuda+cudnn+pytorch深度学习环境的一个例子来简单直白的说明怎样去做

再提醒一遍,下面的操作也要在你激活自己的虚拟环境的情况下进行!

三、安装CUDATookit

我们需要用到下面这条命令

conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

我选择下载11.3版本,这个需要看你们自己的需求来改变,只要你系统的CUDA支持就可以

四、安装cuDNN

如果你成功安装了你想要的那个版本的cudatookit,注意,现在你安装cudnn的版本必须依赖于cudatookit的版本

cuda与cudnn的对应关系可以在cuDNN历史版本下载页面看到:

  • cuda下载页面
  • cudnn下载页面

这里我简单列出来了较新的一些版本之间的对应关系

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

根据上图可知,我安装了CUDA 11.3版本,那么可选的cuDNN版本有很多,这里我直接无脑安装了最新版本,也就是cuDNN的8.4.0版本,同样是用清华镜像源来安装

conda install cudnn=8.4.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

五、安装Pytorch

根据之前我们讲过的,pytorch的版本也是和CUDA版本有对应关系的,比如torch1.6.0只适配cuda10.2、10.1、9.2,不适配cuda11.0。

我们接下来的操作需要进入到Pytorch的官网Previous PyTorch Versions | PyTorch,在里面查看你想要的Pytorch版本它适配的CUDA版本并获取安装命令

比如我现在想要安装Pytorch的1.11.0版本,同时我之前已经安装了CUDA的11.3版本

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)
如上图,这一条命令可以满足我的pytorch和cuda之间的对应关系,因此,我们复制它并运行,即可安装Pytorch 1.11.0

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

注意:

  1. 也可以用pip install的命令来下载pytorch,但是因为三四两步都用的是conda,所以这里为了方便也用了conda
  2. conda install pytorch安装的是torch CPU版本,conda install pytorch torchvision -c pytorch安装的是GPU版本

检查环境是否配置成功

如果如下操作可以正常进行并打印出你安装的相应版本,那么你已经配置成功

#进入虚拟环境
conda activate [你的虚拟环境名]

#输入python来进入python的环境
python

#加载torch
import torch

print(torch.backends.cudnn.version())
#输出8200,代表着成功安装了cudnn v8.4.0

print(torch.__version__)
#输出1.11.0,代表成功安装了pytorch 1.11.0

print(torch.version.cuda)
#输出11.3,代表成功安装了cuda 11.3

torch.cuda.is_available()
#True

码字不易,大家点赞支持一下,感谢!!!文章来源地址https://www.toymoban.com/news/detail-414034.html

到了这里,关于在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • ubuntu20.04配置ros noetic和cuda,cudnn,anaconda,pytorch深度学习的环境

    这里介绍下本篇文章的目的,为了方便自己日后在其他主机上搭建环境,也为了帮助遇到相同问题的人。本篇文章主要是解决ubuntu20.04搭建机械臂视觉抓取的环境部署问题。第一个环境了ROS环境,第二个环境是深度学习yolov5的环境。 这里推荐鱼香ros的便携式安装方法,这里感

    2024年02月07日
    浏览(55)
  • 深度学习环境安装|PyCharm,Anaconda,PyTorch,CUDA,cuDNN等

    本文参考了许多优秀博主的博客,大部分安装步骤可在其他博客中找到,鉴于我本人第一次安装后,时隔半年,我忘记了当时安装的许多细节和版本信息,所以再一次报错时,重装花费了大量时间。因此,我觉得有必要把主要过程记录下来,以便下次需要时快速安装。以下过

    2024年02月17日
    浏览(47)
  • PyTorch深度学习环境安装(Anaconda、CUDA、cuDNN)及关联PyCharm

    Tytorch : Python 机器学习库,基于 Torch ,用于自然语言处理等应用程序 Anaconda :是默认的 python 包和环境管理工具,安装了 anaconda ,就默认安装了 conda CUDA : CUDA 是一种由显卡厂商 NVIDIA 推出的通用并行计算架构,该架构使 GPU 能解决复杂的计算问题,可用来计算深度学习 c

    2024年02月13日
    浏览(45)
  • Ubuntu22.04搭建Pytorch框架深度学习环境+安装Miniconda+安装CUDA与cuDNN

    以Ubuntu为例,从零搭建Pytorch框架深度学习环境。 1.1 系统下载 访问地址ubuntu官网 1.2 启动盘制作 访问ultraiso官网 1.2.1打开镜像 1.2.2写入镜像 1.3磁盘分区 1.3.1Windows磁盘管理 对要压缩的卷右键,选择压缩卷 压缩出的可用空间不要进行分区,等待ubuntu系统操作 1.3.2分区助手或DG

    2024年02月02日
    浏览(51)
  • 深度学习—Python、Cuda、Cudnn、Torch环境配置搭建

    近期由于毕设需要使用Yolo,于是经过两天捣腾,加上看了CSDN上各位大佬的经验帖后,成功搭建好了GPU环境,并能成功使用。因而在此写下这次搭建的历程。 万事开头难,搭建环境很费时间,如果一开始版本不对应,到后面就要改来改去,很麻烦。首先要注意以下事项: 1.

    2024年02月11日
    浏览(30)
  • 【autodl/linux配环境心得:conda/本地配cuda,cudnn及pytorch心得】-未完成

    都是因为国内的网络环境以及墙的问题,导致我们服务器使用了源,或conda使用了源的问题。 使用命令: conda install ,提示找不到包,很可能是因为我们使用了源,然后源里没存那个包。 使用命令: conda search cudatoolkit,找不到包,那其实是这个源里没这个包,不是说conda真的

    2024年02月09日
    浏览(26)
  • 配置Cuda及Cudnn,在Anaconda创建虚拟环境,安装GPU版Pytorch,并在Jupyter noterbook及Pycharm中调用【极其详细】

    目录  一、配置Cuda及Cudnn (一)下载Cuda 1、查看电脑自带的Cuda版本 2、下载相应版本Cuda安装包 3、安装Cuda 4、配置变量 (二)下载Cudnn 二、创建虚拟环境 三、虚拟环境中安装GPU版Pytorch (一)有关环境的基本指令 (二)安装GPU版Pytorch 四、将虚拟环境在Jupyter noterbook及Pycha

    2024年04月28日
    浏览(36)
  • 【autodl/linux配环境心得:conda/本地配cuda,cudnn,gcc,g++及pytorch心得】

    都是因为国内的网络环境以及墙的问题,导致我们服务器使用了源,或conda使用了源的问题。 使用命令: conda install ,提示找不到包,很可能是因为我们使用了源,然后源里没存那个包。 使用命令: conda search cudatoolkit,找不到包,那其实是这个源里没这个包,不是说conda真的

    2024年02月04日
    浏览(26)
  • Halcon22.11+Win10+RTX3060显卡(配置CUDA和cuDNN深度学习环境)

    官网下载安装包(需要注册才能下载) 提示:它和VS2019联合开发,所以要安装VS2019  具体安装步骤参考:https://zhuanlan.zhihu.com/p/590352632   相关的安装包比如CUDA、cudnn版本需要下载到本地安装。  CUDA版本选择,已本人的为例:在NVIDIA控制面板可以看到RTX3060驱动目前最高支持

    2024年02月16日
    浏览(38)
  • Linux 的 anaconda 虚拟环境下安装指定的 cuda、cudnn、pytorch

    感悟:首先,anaconda 的虚拟环境真香!开辟一个新的虚拟环境,很多环境、版本不兼容的问题都不复存在,尤其对复现别人代码的同学很有用。 条件:只要安装的版本不超过自己机器的硬件条件,那么就可以安装。 1. 确定安装的 cuda 版本。 在虚拟环境中,先用 conda search c

    2024年02月09日
    浏览(50)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包