文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

关于PyTorch环境配置及安装教程(Windows10)

2024-04-02 19:55

关注

新手在配置pytorch过程中总会或多或少遇到些问题,同时网上关于pytorch的环境配置琳琅满目,不知道应该按照哪个配置,这里笔者记录一下自己在windows10下配置Pytorch的全过程。

笔者电脑环境以及安装版本为:Windows10 企业版 + python 3.7.9 + Anaconda3 + GTX 1050显卡 + pytorch 1.7.1

一、Anaconda安装

这里推荐大家安装Anaconda(但不是必须)。因为有些时候大家编程的python版本可能不同,所以一般大型项目建议一个项目使用一个python interpreter(编辑器),方便管理,而Anaconda就很方便的帮助我们做这件事情。

安装过程非常简单,在官网找到对应版本安装即可,官网安装页面如下:https://www.anaconda.com/products/individual#Downloads

开始菜单中打开Anaconda中的Anaconda Prompt,出现(base),即安装成功,如下:

Anaconda可以很方便的管理我们我们的所有运行环境以及环境中的各种依赖包,主页面如下:

二、安装pytorch环境

1. 停留在上面 Anaconda Prompt 页面上

输入

conda create -n environment_name python=3.7

因为我们之前说过,我们需要给每个项目配置一个python编辑器,所以environment_name(改成你自己指定的环境名,比如就叫environment_test)就是我们配置的这个项目对应的编译环境的名字,里面放入我们指定的python编辑器(这里我以python 3.7为例)。

2. 出现上述页面

输入y,表示确认安装上述所需要的包。

出现上图,环境中的python编辑器安装完成。

当然现在我们还处于(base)环境,也就是Anaconda自带的基础环境,我们需要跳到我们刚刚创建的environment_test环境。

3. 按照上图

执行conda activate environment_test,跳到environment_test编译环境中

此时环境已经发生改变,输入pip list,可以看到目前环境中所有已经安装的包,但是没有我们想要的pytorch,需要我们自己安装。

4. pytorch安装

pytorch安装非常的人性化,我们只需要去pytorch官网,点击install,就可以看见下面这个界面:

对于每一项的选择在这里做一下详细介绍:

注意CUDA9.2以上版本支持的Driver Version不可以低于396.26,一般建议安装前先更新一下自己显卡的驱动,笔者昨天刚刚更新完,直接点更新,很方便。

注意:python 3.9 还需要在结尾增加-c=conda-forge

可以看到里面包括pytorch,numpy等我们所需要的一些包,还是输入y,等待安装。

⑦ 出现done,安装完成 证明pytorch安装成功,关闭当前Anaconda Prompt,并重新从开始菜单打开,并激活environment_test环境,在环境下打开python编辑页面。

输入import torch,不报错证明pytorch包安装成功。

输入torch.cuda.is_available(),显示True,证明cuda安装成功,可以使用GPU对pytorch中的张量、网络进行加速处理。若为False,此时pytorch只能用CPU运行。

三、可能遇到的一些问题

笔者在这里记录第一次安装时出现的一些问题,在这里记录一下~

1. pytorch包安装不成功的话,建议按照提示更新一下conda版本(可能是你的Anaconda版本太老了,新安装的应该不会出现这个问题)

2.‘nvidia-smi’ 不是内部或外部命令:解决方法如下

3. 环境中安装jupyter,报react404之类的错误:解决方法如下

[W xx:xx:xx.xxx NotebookApp] 404 GET/static/components/react/react-dom.production.min.js (::1)

pip install --user --ignore-installed jupyter

4. 上述的第5步证明pytorch安装成功,一定要关闭当前Anaconda Prompt,再重新打开激活环境,因为新安装完之后环境是未激活状态,这时候一些依赖包是不能被调用的

5. torch.cuda.is_available()显示false的几个原因:

① 未正确安装GPU,查看 任务管理器–性能,下面是否有你的显卡 

② CUDA9.2以上版本支持的Driver Version不可以低于396.26,一般建议安装前先更新一下自己显卡的驱动。不建议去英伟达官网更新,因为查询最新的驱动有很多,不知道会安装哪个。直接英伟达显卡设置管理软件中安装即可,见上文。

③ 安装的CUDA版本不能高于显卡支持的CUDA版本。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持编程网。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯