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
    +关注

    关注

    28

    文章

    5100

    浏览量

    134477
  • 深度学习
    +关注

    关注

    73

    文章

    5591

    浏览量

    123915
  • pytorch
    +关注

    关注

    2

    文章

    813

    浏览量

    14706
收藏 人收藏
加入交流群
微信小助手二维码

扫码添加小助手

加入工程师交流群

    评论

    相关推荐
    热点推荐

    停电接线的电能质量在线监测装置的安装和调试复杂吗?

    停电接线的电能质量在线监测装置 整体安装调试复杂度较低 ,远低于传统停电接线模式,其中低压系统可实现 “快速部署、简易调试”,中高压系统因需专业工具与资质,复杂度略有提升,但仍能通过标准化流程降低
    的头像 发表于 12-05 18:00 3327次阅读
    <b class='flag-5'>免</b>停电接线的电能质量在线监测装置的<b class='flag-5'>安装</b>和调试复杂吗?

    ubuntu系统安装之Vmware虚拟机安装

    ,装到自己电脑安装软件的分区,点击“下一步”。 勾选,点击“下一步”。 勾选添加快捷方式,点击“下一步”。 点击“安装”。 等待安装完成。 点击完成安装。双击桌面上的VMwa
    发表于 09-22 16:27

    KiCad 安装版本(不带器件库瘦身版)

    个特殊的版本,同时解决这两个问题。  ”   什么是安装版? 顾名思义,就是下载之后能直接用,不用安装的操作。除此之外,安装包还进行了瘦身,把所有的原理图符号、封装以及 3D 模型(
    的头像 发表于 09-02 11:20 1w次阅读
    KiCad <b class='flag-5'>免</b><b class='flag-5'>安装</b>版本(不带器件库瘦身版)

    焊BNC接头接法:零基础快速安装指南与信号稳定技巧

    通过以上步骤,即使零基础用户也能快速完成焊BNC接头的安装,确保视频、音频信号稳定传输,满足监控、音视频工程等场景需求。
    的头像 发表于 08-28 16:59 777次阅读
    <b class='flag-5'>免</b>焊BNC接头接法:零基础快速<b class='flag-5'>安装</b>指南与信号稳定技巧

    Molex简化户外安装的Quasar OptiX现场安装连接器在贸泽开售

    安装连接器。此系列现场安装连接器让安装人员能够轻松将工厂抛光连接器和机械式接头连接器安装到受保护的户外环境中的光缆上。Quasar Opti
    发表于 07-31 15:26 612次阅读
    Molex简化户外<b class='flag-5'>安装</b>的Quasar OptiX现场<b class='flag-5'>安装</b>连接器在贸泽开售

    安装教程】LP20系列USB工业级连接器安装详解

    /USB3.0可选,满足高速传输需求;④品质电缆组件,接线,安装更省心;⑤按扣直插式连接设计,单手插拔更快捷。1LP20-USB连接器安装流程介绍01安装插座该款产
    的头像 发表于 04-25 18:09 839次阅读
    【<b class='flag-5'>安装</b>教程】LP20系列USB工业级连接器<b class='flag-5'>安装</b>详解

    人脸识别指南:如何在树莓派上安装和设置 Dlib

    学习如何在树莓派上安装Dlib并配置人脸识别功能,为您的AI项目奠定基础。在树莓派上安装Dlib的详细步骤要为树莓派安装Dlib并确保人脸识别项目顺利运行,请按照以下详细步骤操作。前置
    的头像 发表于 03-24 17:31 1262次阅读
    人脸识别指南:如何在树莓派上<b class='flag-5'>安装</b>和设置 Dlib

    【ELF 2学习板试用】利用RKNN-Toolkit2实现rknn模型转换

    版本,参考下面这篇blog即可VMware虚拟机安装Ubuntu教程(超详细)_vmware安装ubuntu-CSDN博客,也可以自己去找攻略来进行配置。 开发环境准备 正常ubuntu系统是x86
    发表于 02-09 17:57

    操作指南:pytorch云服务器怎么设置?

    设置PyTorch云服务器需选择云平台,创建合适的GPU实例,安装操作系统、Python及Anaconda,创建虚拟环境,根据CUDA版本
    的头像 发表于 02-08 10:33 609次阅读

    华为云 Flexus X 实例评测使用体验——Anaconda 环境安装

    Anaconda 是 python 开发最常用环境,那么我们今天使用【华为云 Flexus X 实例】来搭建一下这个环境,希望本文能对 python 开发者们有一定的价值,当然,这里也推荐一下华为的【CodeArts】用着的确是
    的头像 发表于 01-21 16:13 691次阅读
    华为云 Flexus X 实例评测使用体验——<b class='flag-5'>Anaconda</b> <b class='flag-5'>环境</b><b class='flag-5'>安装</b>

    隔离变压器的安装注意事项

    隔离变压器的安装是一个重要的环节,以下是一些关键的安装注意事项: 选择合适的安装位置 : 隔离变压器应安装在干燥、通风良好的场所,避免潮湿、高温、多尘或有腐蚀性气体的
    的头像 发表于 01-07 09:56 1638次阅读

    华为云 Flexus X 实例部署安装 Jupyter Notebook,学习 AI,机器学习算法

    前言 由于本人最近在学习一些机器算法,AI 算法的知识,需要搭建一个学习环境,所以就在最近购买的华为云 Flexus X 实例上安装学习
    的头像 发表于 01-02 13:43 867次阅读
    华为云 Flexus X 实例部署<b class='flag-5'>安装</b> Jupyter Notebook,<b class='flag-5'>学习</b> AI,机器<b class='flag-5'>学习</b>算法

    工业连接器的安装步骤详解

    在现代工业自动化和控制系统中,工业连接器扮演着至关重要的角色。它们不仅需要确保信号的可靠传输,还要能够承受极端的温度、湿度、振动和腐蚀等环境因素。因此,正确安装工业连接器对于确保设备长期稳定运行
    的头像 发表于 12-29 10:16 1688次阅读

    电脑安装联想私有云,电脑如何安装联想私有云?

        在局域网环境中,云电脑工具是企业实现高效内部管理和协作的重要工具。这些软件不仅能够帮助企业提升工作效率,还能增强信息安全和监控能力。今天小编要讲解电脑如何安装联想私有云。    在电脑上安装
    的头像 发表于 12-25 13:34 1278次阅读
    电脑<b class='flag-5'>安装</b>联想私有云,电脑如何<b class='flag-5'>安装</b>联想私有云?

    Triton编译器安装步骤详解

    :用于构建项目。 Python :用于运行 Triton 的 Python 绑定。 其他依赖 :根据您选择的架构,可能需要额外的依赖。 2. 安装依赖 对于 Linux: 打开终端并运行以下命令来安装
    的头像 发表于 12-24 17:35 4636次阅读