0
  • 聊天消息
  • 系统消息
  • 评论与回复
登录后你可以
  • 下载海量资料
  • 学习在线课程
  • 观看技术视频
  • 写文章/发帖/加入社区
创作中心

完善资料让更多小伙伴认识你,还能领取20积分哦,立即完善>

3天内不再提示

基于Anaconda安装pytorch深度学习环境+pycharm安装---免额外安装CUDA和cudnn

成都华江信息 2023-10-10 10:16 次阅读

前言

最近由于项目需要,之前我们在利用GPU进行深度学习的时候,都要去NVIDIA的官网下载CUDA的安装程序和cudnn的压缩包,然后再进行很繁琐的系统环境配置。不仅环境配置麻烦,而且还特别容易配置错误,特别还有CUDA和cudnn版本的对应也特别容易搞错,但是利用anaconda安装配置pytorch和paddle环境的时候会自动帮我们配置好cuda和cudnn。这篇文章可以帮助小白快速配置深度学习环境。

一、NVIDIA驱动安装

显卡驱动程序就是用来驱动显卡的程序,它是硬件所对应的软件。驱动程序即添加到操作系统中的一小块代码,其中包含有关硬件设备的信息。正常有显卡的电脑都是有驱动程序的,但是有的时候驱动可能版本比较低,支持的cuda版本也是比较低的(但是有的人的显卡是比较老的,就不建议更新驱动,这样会导致各种各样的问题,但是搞深度学习还是要用一块好的显卡用来学习,这点是过来人有血泪教训的,如果显卡性能强悍,可以大大节省AI模型训练时间,提高效率。)

首先查看电脑的显卡版本,步骤为:此电脑右击-->管理-->设备管理器-->显示适配器。就可以看到电脑显卡的版本了。09eac842-6713-11ee-9788-92fbcf53809c.png

有显卡驱动的,可以直接在桌面右键,找到英伟达驱动控制面板打开就好了。 0a082c16-6713-11ee-9788-92fbcf53809c.png

在得知以上的信息以后我们就可以对应我们的显卡去英伟达官网上去找相对应的显卡驱动更新或者下载了。

显卡驱动的下载地址(可能打开比较慢,多打开几遍)。

然后根据自己的电脑的配置去选择驱动。这里Notebooks是笔记本的意思,所以如果你是笔记本电脑,那么产品系列那个选项就要选(Notebooks)的。还有那个下载类型有两种一个是Studio版本,一个是Game Ready版本。其实两个版本都差不多,一个是偏办公用,一个是偏游戏娱乐。按如下操作将驱动下载下来。然后点击下载来的程序,不断的下一步就好了。

安装(更新)好了显卡驱动以后。我们按下win+R组合键,打开cmd命令窗口。输入如下的命令。

nvidia-smi

得到如下图的信息图,可以看到驱动的版本是442.50;最高支持的CUDA版本是10.2版本。得到显卡的最高支持的CUDA版本,我们就可以根据这个信息来安装环境了。0a59a6ae-6713-11ee-9788-92fbcf53809c.png

二、Anaconda 的安装

打开网址,现在是2021年10月,对应的anaconda版本是支持python3.8。如果想下载之前的版本,或者更低python版本的anaconda,可以打开网址。0a692bf6-6713-11ee-9788-92fbcf53809c.png 双击下载好的anaconda安装包,just me是说只供当前用户使用。all user 是供使用这台电脑的所有用户使用,是权限问题。对空间影响不大。如果你的电脑上只有建了一个用户,all users和just me 的作用是一样的。所以点击just me就好了。0a7b5f1a-6713-11ee-9788-92fbcf53809c.png 然后点击next,当让你选择安装安装路径的时候,一定不要选择默认安装位置,因为默认位置是c盘,以后要在anaconda里面创建环境的时候会很占内存,最好在D盘中创建一个文件夹来放anaconda。0a89889c-6713-11ee-9788-92fbcf53809c.png 和图中一样将图中的√勾上,虽然出现红色的警告,但是要勾上,将anaconda添加到环境变量中去。然后点击完成就好了。0aa1855a-6713-11ee-9788-92fbcf53809c.png 安装完成以后,按下开始键(win键)在左边就会出现anaconda3这个文件夹,可以发现anaconda已经安装好了。0abdb630-6713-11ee-9788-92fbcf53809c.png

三、Pytorch环境安装

按下开始键(win键),点击如图中的图标。打开anaconda的终端。

0ad682c8-6713-11ee-9788-92fbcf53809c.png       执行如下的指令查看有哪些环境

condaenvlist

可以看出来,新安装的anaconda只有一个base环境0aeaddc2-6713-11ee-9788-92fbcf53809c.png 这里先讲一下anaconda环境,首先base环境是一个大的环境,类似一个很大的一个房子(但是没有房间),当我们每创建一个环境就都会相当于在这个大房子里面用隔板创建一个房间,然后这个房间里面可以安装我们所需要的包,这样管理起来就比较方便。如图可以比较直观的诠释anaconda的环境 0af7ed82-6713-11ee-9788-92fbcf53809c.png 创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下,我就是创建了一个名字叫pytorch,python是3.8版本的环境。

condacreate-npytorchpython=3.8

在base环境中执行如上的命令,就会创建一个新的虚拟环境,这个虚拟环境会安装一些基础的包,如下图所示。询问是否安装的时候,输入y。就可以创建环境了。0b0a7f4c-6713-11ee-9788-92fbcf53809c.png 当安装好了以后,执行conda env list这个命令,就可以看到比一开始多了一个pytorch这个环境。现在我们可以在这个环境里面安装深度学习框架和一些Python包了。0b27f5b8-6713-11ee-9788-92fbcf53809c.png 执行如下命令,激活这个环境。conda activate 虚拟环境名称

condaactivatepytorch

安装pytorch-gup版的环境,由于pytorch的官网在国外,下载相关的环境包是比较慢的,所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源。

condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
condaconfig--setshow_channel_urlsyes

0b55a51c-6713-11ee-9788-92fbcf53809c.png       然后打开pytorch的官网,由于开头我们通过驱动检测到我的显卡为 GeForce MX350,最高支持cuda10.2版本,所以我们选择cuda10.2版本的cuda,然后将下面红色框框中的内容复制下来。

0b6d2336-6713-11ee-9788-92fbcf53809c.png       将复制的内容粘贴到pytorch环境下的终端,运行就可以了

condainstallpytorch==1.12.1torchvision==0.13.1torchaudio==0.12.1cudatoolkit=10.2-cpytorch

这时候就开始下载环境所需要的依赖包了。

四、pycharm安装--验证CUDA和cudnn版本

打开这个pycharm网址,可以发现一共有两个版本一个是专业版(Professional),一个是社区版(Community),专业版是需要花钱的,好几百美元一年。而社区版是免费的,但是也够用了,所以就下载安装社区版就好了。0b85d674-6713-11ee-9788-92fbcf53809c.png       将下载好的pycharm安装包,双击。安装的地址最好不要放在C盘,我是安装在D盘(反正我 的软件是不喜欢安装在C盘的。总害怕C盘会满)。0b988f44-6713-11ee-9788-92fbcf53809c.png       将所有的√都勾上,0bad733c-6713-11ee-9788-92fbcf53809c.png 安装好了点击第二个框框,然后点完成就好了0bbf804a-6713-11ee-9788-92fbcf53809c.png 打开pycharm,创建一个新的工程,来测试anaconda是否在安装pytorch和paddlepaddle框架的时候也安装了cuda和cudnn。按如下两图创建一个工程(新工程好像必须要安装一个新的python插件),新的工程最好在D盘一个新的文件夹下,有的工程很大,C盘容易装满。0bccc980-6713-11ee-9788-92fbcf53809c.png0be02e1c-6713-11ee-9788-92fbcf53809c.png 按以上的方式创建了一个工程,这时候我们就要选择我们在anaconda里面安装的环境,在界面的右下角0bfc3da0-6713-11ee-9788-92fbcf53809c.png 按照如图中的选项选择我们在anaconda中创建的深度学习环境,可以看到有pytorch环境。0c0de6fe-6713-11ee-9788-92fbcf53809c.png

此时刚刚的右下角已经有了我们刚刚选择的pytorch环境中的python了。0c2a7f80-6713-11ee-9788-92fbcf53809c.png      我们在创建的工程里面创建一个python脚本,在脚本中运行如下代码,查看是否anconda在安装pytorch环境的时候也安装了cuda和cudnn。

importtorch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

可以发现控制台打印出两个True,可以说明cuda和cudnn已经安装。并且可以得到cuda的版本为10.2和cudnn的版本为7.6.5版本。0c401dfe-6713-11ee-9788-92fbcf53809c.png

至此我们的深度学习环境安装就已经完全完成。接下来可以在相应的环境下进行深度学习的实验了。

最后还要申明一下,你可以创建不同的环境,在里面安装不同版本的cuda和cudnn版本。已经亲自尝试过了,是可以的。

声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉
  • gpu
    gpu
    +关注

    关注

    27

    文章

    4419

    浏览量

    126705
  • 深度学习
    +关注

    关注

    73

    文章

    5239

    浏览量

    119910
  • pytorch
    +关注

    关注

    2

    文章

    762

    浏览量

    12835
收藏 人收藏

    评论

    相关推荐

    PyTorch安装教程超详细

    PyTorch是一个用于机器学习深度学习的开源库,它提供了丰富的工具和接口,帮助开发者快速构建深度学习
    的头像 发表于 12-07 11:19 1156次阅读

    python运行环境安装和配置

    Python是一种非常流行的编程语言,广泛应用于科学计算、Web开发、人工智能等领域。为了能够正常运行Python程序,我们需要先安装和配置Python运行环境。本文将为您详尽介绍Python运行
    的头像 发表于 11-29 16:17 544次阅读

    32位的win7如何安装pycharm

    安装PyCharm在32位的Windows 7系统上是可能的,虽然32位系统的支持已经逐渐减少,但PyCharm仍然提供可用于此系统的安装程序。在本文中,我将详细介绍在32位的Wind
    的头像 发表于 11-23 17:09 2076次阅读

    Pycharm+Django安装及配置

    给其提供了配置模板: 能直接在新建项目的时候选择Django并新建一个独立的虚拟环境: 从新建到编码测试,一套流程用起来都极其方便。 1.下载Pycharm 在jetbrains官网选择相应的系统下载
    的头像 发表于 11-02 11:28 546次阅读
    <b class='flag-5'>Pycharm</b>+Django<b class='flag-5'>安装</b>及配置

    Qlib的安装和运行内置算法策略

    )》或 《 Python数据分析与挖掘好帮手—Anaconda 》 进行安装,本文建议使用Anaconda。 由于qlib有许多许多依赖,如果你不想安装过程中出现问题,或者引起其他程
    的头像 发表于 10-31 10:19 2468次阅读
    Qlib的<b class='flag-5'>安装</b>和运行内置算法策略

    请问ubantu如何确定安装好了cudnn?

    ubantu如何确定安装好了cudnn?​
    发表于 10-20 06:35

    Newspaper开源模块的安装和使用教程

    这篇文章: 超详细Python安装指南 进行安装。 如果你用Python的目的是数据分析,可以直接安装Anaconda: Python数据分析与挖掘好帮手—
    的头像 发表于 10-16 11:22 292次阅读

    安装Linux的系统安装工具

    安装_Linux的系统安装工具,可以安装各类Linux操作系统。通过iso镜像文件的格式安装。也可以安装在U盘上,本人亲测通过。  
    发表于 09-11 10:21 0次下载

    深度学习框架pytorch入门与实践

    深度学习框架pytorch入门与实践 深度学习是机器学习中的一个分支,它使用多层神经网络对大量数
    的头像 发表于 08-17 16:03 1201次阅读

    OpenPCDet环境配置

    ,这样会导致出一些额外的错误。我参考的是这篇博客进行cmake的升级。 1.1.2 验证cudacudnn是否安装好了 先说一下我遇到的问题,因为这次的机子是我接手别人的,里边看似是
    的头像 发表于 08-07 11:06 932次阅读
    OpenPCDet<b class='flag-5'>环境</b>配置

    安全光幕对安装环境有什么要求?

    安全光幕对安装环境有什么要求
    的头像 发表于 06-27 15:23 279次阅读

    PyTorch教程5.5之深度学习中的泛化

    电子发烧友网站提供《PyTorch教程5.5之深度学习中的泛化.pdf》资料免费下载
    发表于 06-05 15:31 1次下载
    <b class='flag-5'>PyTorch</b>教程5.5之<b class='flag-5'>深度</b><b class='flag-5'>学习</b>中的泛化

    PyTorch教程12.1之优化和深度学习

    电子发烧友网站提供《PyTorch教程12.1之优化和深度学习.pdf》资料免费下载
    发表于 06-05 15:08 0次下载
    <b class='flag-5'>PyTorch</b>教程12.1之优化和<b class='flag-5'>深度</b><b class='flag-5'>学习</b>

    使用pycharm开发上位机配置pyqt5的环境

    在上位机的开发过程中,使用QT的图形化界面,是开发上位机最好最简单的选择,因此我将讲述配置pyqt5的环境。 软件安装安装pycharm的社区2021.02版 QT(
    发表于 05-29 16:12 0次下载
    使用<b class='flag-5'>pycharm</b>开发上位机配置pyqt5的<b class='flag-5'>环境</b>

    pycharm写上位机

    pycharm上写上位机 安装pyqt5(安装好了Anaconda)用pip安装: pip install pyqt5,
    发表于 05-09 14:27 0次下载
    用<b class='flag-5'>pycharm</b>写上位机