当前位置:   article > 正文

在windows系统中安装pytorch(保姆级教程)_windows安装pytorch

windows安装pytorch

首先需要在你的电脑中安装anaconda系列,本人用的是miniconda,大家在官网搜索下载即可

以下为miniconda下载教程

在bing浏览器搜索miniconda官网后,进入官网(界面如下)

 在此界面下滑,找到你的系统及对应python版本(笔者的是python3.8版本,电脑是64位)

 点击蓝色字体开始下载

下载好后点击电脑开始界面,找到anaconda prompt点开,在命令行输入以下代码:

conda create -n env_name xxx(此处填写你想要的虚拟环境的名称,并非必须写xxx)

创建好后运行以下代码激活虚拟环境:

conda activate env_name

如果是第一次使用anaconda,笔者建议继续按现在的教程进行,如果之前已用过,笔者建议跳到安装pytorch这一步 

激活后退出虚拟环境,运行以下代码:

conda deactivate

之后打开pycharm,选择新建项目,位置选择为刚刚新建的虚拟环境,路径为下图所示,笔者事先在D盘创建了文件夹miniconda存放miniconda的文件,虚拟环境名设为了YOLO,对应修改即可

 建好项目后再次找到anaconda prompt点开,在命令行输入代码(忘记的话上滑)激活虚拟环境,之后打开电脑NVIDIA控制面板查找自己的显卡信息,方法如下:

点击后依次点击“帮助”——>“系统信息”——>“组件” 

 红框选中部分可以看到我的CUDA是12.0版本,大家要记住自己的版本

之后bing搜索pytorch官网

在顶部这一栏选择Get Started,之后下滑,进行如图选择

红色框内Compute Platform部分向下兼容,只要选择比自己CUDA版本低的就可以

选择好后蓝色框内会出现对应代码,将出现的代码复制粘贴到 anaconda prompt中Enter运行即可,在出现的问题中选择y

 以下为安装完成的界面,过程需要一定时间,大家耐心等待即可

同时我们在pycharm中也可以看到pytorch安装完毕

以上便是笔者关于pytorch安装的分享,大家在操作过程中如果有什么疑问,欢迎评论留言 

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/967503
推荐阅读
相关标签
  

闽ICP备14008679号