深度学习—Python、Cuda、Cudnn、Torch环境配置搭建

这篇具有很好参考价值的文章主要介绍了深度学习—Python、Cuda、Cudnn、Torch环境配置搭建。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

近期由于毕设需要使用Yolo,于是经过两天捣腾,加上看了CSDN上各位大佬的经验帖后,成功搭建好了GPU环境,并能成功使用。因而在此写下这次搭建的历程。

万事开头难,搭建环境很费时间,如果一开始版本不对应,到后面就要改来改去,很麻烦。首先要注意以下事项:

1. 你的显卡驱动版本。

2. 你的显卡算力。

3. Cuda和Cudnn版本对应问题。

4. Torch和Python对应关系。

我个人配置如下:

Python 3.7 + CUDA 11.6 + CUDNN 8.4.0 + Torch 1.12.0


目录

一、Anaconda下载与安装

二、查看电脑显卡信息

tips:查看显卡算力

三、CUDA下载与安装

3.1 查询Cuda版本

3.2 下载CUDA

3.3 CUDA安装

3.4 检验CUDA 是否安装成功

四、CUDNN下载与安装

4.1 CUDNN下载

4.2 CUDNN解压

4.3 检测CUDNN是否安装成功

五、Pytorch安装

5.1 创建Anaconda虚拟环境

5.2 下载Torch

5.3 检验torch是否安装成功

参考致谢


错误示范:

刚开始我的CUDA用的是11.4,后面下载Torch的时候报错,找不到满足版本的torch。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习


一、Anaconda下载与安装

下载地址:Anaconda | The World's Most Popular Data Science Platform

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

二、查看电脑显卡信息

首先我们得查看自己电脑是否有独立显卡。打开设备管理器 -> 显示适配器。可以看到我的电脑上有两块显卡。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

其次,打开NVIDIA控制面板,查看显卡信息。右键 -> NVIDIA控制面板 -> 系统信息 -> 组件

记住你的驱动程序版本和你的NVCUDA64.DLL对应的版本名称。后续要用。

注意高版本的Cuda是可以向下兼容的,但是不支持低版本向上兼容。我的电脑支持CUDA11.6,因此我可以向下下载cuda11.3等等,但是比如说你的是CUDA 11.0的话,你就不能下载CUDA 11.3和CUDA 11.6。这是跟着你电脑显卡信息来的。切记!!!

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

tips:查看显卡算力

打开NVIDIA网站:CUDA GPU | NVIDIA Developer

找到相应显卡所在区域,点击查看你的显卡算力。cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

三、CUDA下载与安装

3.1 查询Cuda版本

打开网址:CUDA 12.1 Release Notes

下滑至该表格处,windows看右边的驱动版本。表格中CUDA 11.6.x 要>=452.39。我的电脑是512.36(第二步中要记下来的那个驱动版本号),满足条件。cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

3.2 下载CUDA

打开网址:CUDA Toolkit Archive | NVIDIA Developer

找到对应的CUDA版本,下载。 强烈建议用迅雷!!!

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

3.3 CUDA安装

打开exe文件,地址默认不要改。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

进入安装,然后选择自定义(高级 ),下一步就行了。[该安装过程忘了截图了。]

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

3.4 检验CUDA 是否安装成功

按 win + R ,输入cmd,进入命令行,输入命令: nvcc -V   (注意中间有空格)

 nvcc -V 

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

四、CUDNN下载与安装

4.1 CUDNN下载

打开网址:CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

找到对应CUDA 版本的CUDNN版本,下载。下载时需要登录NVIDIA账号,注册即可。下载还是建议用迅雷。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

不要选择太新的CUDNN,我自己下的是CUDNN 8.4.0cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

4.2 CUDNN解压

CUDNN是一个压缩包,需要解压到相关目录里。

打开CUDA所在目录,我自己电脑上的是 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

 CUDNN压缩包里的内容是三个文件夹。将这三个文件夹覆盖到CUDA\v11.6的同名文件夹即可。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

4.3 检测CUDNN是否安装成功

按 win + R,输入cmd,切换目录路径到CUDA文件夹里的 v11.6 下的 extras 文件夹下的demo_suite文件夹中,按顺序执行deviceQuery.exebandwidthTest.exe文件。

注意文件位置!!!

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

 执行deviceQuery.exe, 最后显示PASS。cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

 执行bandwithTest.exe, 最后显示PASS。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

五、Pytorch安装

5.1 创建Anaconda虚拟环境

打开Anaconda Prompt ,输入命令:conda create -n 你创建的环境姓名 python==版本号

conda create -n pytorch python==3.7

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习
等待安装好后,输入命令: activate  你创建的环境名称 ,激活虚拟环境。
activate pytorch

5.2 下载Torch

Torch和Python对应关系:mirrors / pytorch / vision · GitCode

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

打开网址:Previous PyTorch Versions | PyTorch

找到对应的CUDA版本命令输入即可,静静等待。

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

5.3 检验torch是否安装成功

打开Anaconda Prompt ,激活环境,输入python。

注意:因为torch是python当中的,所以要进入python语言环境中。如果直接输入import torch的话就会报错。

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

输入以下代码进行检验

import torch # 如果pytorch安装成功即可导入
print(torch.__version__) #torch版本
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号

cuda和cudnn环境,深度学习,Torch,pytorch,python,深度学习

以上是我这两天来配置环境的过程,希望对正在配置GPU的你有用。如有错误,敬请各位大佬指正。 

参考致谢

[PyTorch] 安装笔记, 基于Windows10/cuda11.6 - 知乎

PyTorch、CUDA Toolkit 及显卡驱动版本对应关系_pytorch cuda版本对应关系_浪迹天涯@wxy的博客-CSDN博客

python、torch、torchvision、CUDA、cudnn、GPU、各驱动版本对应问题记录_torch对应cudnn版本_SigMap的博客-CSDN博客

(conda + pip) 配置各版本 Pytorch 深度学习环境_pytorch版本_听 风、的博客-CSDN博客

Win10安装cuda、cudnn检测是否安装成功_检测cudnn_旅途中的宽~的博客-CSDN博客文章来源地址https://www.toymoban.com/news/detail-677489.html

到了这里,关于深度学习—Python、Cuda、Cudnn、Torch环境配置搭建的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

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

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

相关文章

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

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

    2023年04月15日
    浏览(63)
  • 【新手流程】1小时解决Anaconda+Cuda+Cudnn+Pytorch+Pycharm深度学习环境配置

    1、安装Anaconda 2、安装Cuda+Cudnn 3、安装Pytorch 4、安装Pycharm 5、配置Pycharm环境 点击进入Anaconda官网👉: Anaconda.com官网 https://www.anaconda.com/download/ 如果下载速度太慢可以使用这个镜像链接: Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror Index of /anaconda/a

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

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

    2024年02月16日
    浏览(61)
  • ubuntu20.04配置ros noetic和cuda,cudnn,anaconda,pytorch深度学习的环境

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

    2024年02月07日
    浏览(88)
  • 【深度学习环境配置】ubuntu 20.04+4060 Ti+CUDA 11.8+pytorch(装机、显卡驱动、CUDA、cudnn、pytorch)

    【深度学习环境配置】ubuntu 20.04+4060 Ti+CUDA 11.8+pytorch(装机、显卡驱动、CUDA、cudnn、pytorch) 📆 安装时间 2023.11.08-2023.11.10 Windows 和 Ubuntu 双系统的安装和卸载 B站教程 【本文基本上跟这个详细教程一致,优先推荐看这个!】ubuntu20.04 下深度学习环境配置 史上最详细教程 【精

    2024年02月04日
    浏览(61)
  • pytorch环境配置(装cuda、cudnn)win10+cuda10.1+cudnn7.6.5+torch1.7.1 && 集显装pytorch

    为了装这个走了太多坑了,所以想写一篇具体教程,有缘人看吧,希望能解决你的问题。(第一次写文章啥也不懂,万一冒犯了啥,麻烦告知我改) 我anaconda很早就装过了,所以这里就不细说了。 电脑配置:win10+1050显卡(很久前买的电脑的) 一些弯路:现在的pytorch安装是可

    2024年02月04日
    浏览(53)
  • Ubuntu 20.04 RTX 4090显卡 深度学习环境配置(Nvidia显卡驱动、CUDA11.6.0、cuDNN8.5)

    参考文献:从零到一保姆级Ubuntu深度学习服务器环境配置教程 看文献中“ 三、 NVIDIA驱动安装 ” 安装NVIDIA驱动,这也是安装CUDA10.0及其对应版本的CuDNN和tensorflow的重要步骤。 1.1.1 英伟达中国驱动官网 进入英伟达中国驱动官网 1.1.2 输入显卡型号查询 1.1.3 查看搜索结果 1.2.1 方

    2024年02月04日
    浏览(109)
  • windows11 安装cuda和cudnn深度学习开发环境

    首先先要确认自己的显卡最高能支持到cuda的版本,一般是不限制版本号的。 然后在官网地址下载: cuDNN的官网下载地址:cuDNN Archive | NVIDIA Developer CUDA 的下载地址:CUDA Toolkit Archive | NVIDIA Developer 有一点需要注意的是,需要查看cuda支持的cudnn版本:  找与自己的cuda 版本对于的

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

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

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

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

    2024年02月13日
    浏览(67)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包