当前位置:   article > 正文

pytorch-handbook 国内镜像安装pytorch教程(新手入门)_pytorch handbook

pytorch handbook

安装好 Anaconda 的并且不想听我废话的同学请直接戳:

如何使用国内镜像源下载?


 

没有下载 pytorch 中文手册的请戳 pytorch-handbook github:https://github.com/sktyut/pytorch-handbook

pytorch handbook是一本开源的书籍,目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门,其中包含的Pytorch教程全部通过测试保证可以成功运行

没有下载安装 Anaconda 的请戳:anaconda 的下载安装与conda管理环境

Anaconda的安装十分简单,在此不多赘述,需要注意的就是 Anaconda 自带 python,如果你已经安装了 python 很可能会发生冲突

打开 pytorch-handbook 中文手册(README.md),点击 PyTorch 环境搭建:

然后出现 Pytorch 的安装教程 :

 注意:pytorch-handbook上的教程的前提是已经成功安装好 Anaconda 。然后才能打开 Anaconda Prompt

我们按照教程依次创建环境:(这个环境就是你开发时的虚拟环境,教程这里选择了 python 3.6 版本的开发环境,然后将此环境命名为 pytorch ,可以打开 Anaconda Navigator Appications on 中就能看到你创建的环境了)

conda create -n pytorch python=3.6

然后切换到刚刚创建的环境:

activate pytorch

下一步就是安装 pytorch 了。

pytorch如何在官网选择自己的需要下载的版本

pytorch 有两个版本 分别是 cpu 版和 gpu 版本,由于教程提供的下载命令只是其中一个版本的,所以这里我们打开 pytorch 的官网:https://pytorch.org/

在这里选择你要下载的 pytorch 的版本:

  • PyTorch Build 就是选择下载稳定版(Stable)还是预览版(Preview),这里一般选择稳定版(Stable)
  • YourOS 不多说,选择自己对应的操作系统类型
  • Package:下载方式,教程提供的是 Conda 方式,这里我们使用 Pip 方式
  • Language:即选择你的 python 版本(Anaconda 下载后自带 python ,可以在 cmd 输入 python --version 查看自己的 python 版本)
  • CUDA:如果要下载 gpu 版本,那么就选择你对应的 cuda 的版本(不知道自己 cuda 版本的请戳:https://www.jianshu.com/p/d3b9419a0f89);如果下载 cpu 版本,那么选择 None
  • Run this Command:当你选择好版本后,会自动再次生成下载安装 PyTorch 的命令

这里我选择了 gpu 版本的:


如何使用国内镜像源下载?

只需要在此命令后加上:(这里使用的是清华的镜像站,-i 意思就是指定单次下载的镜像源)

-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

比如我的下载命令就变成了:

pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

然后就可以体验国内资源飞一般的网速了。(如果显示 pip3 报错的,请把3去掉改用 pip 命令)

我们再回到 PyTorch-Handbook 中文教程,下面就是验证自己是否成功安装

很简单还是在自己创建的 pytorch 环境中,输入:

  1. import torch
  2. torch.__version__

能显示出正确的版本号就意味着你已经成功安装了。

下面我们还需要配置 Jupyter Notebook

还是在自己创建的 pytorch 环境中,执行如下命令:

  1. #新建的环境是没有安装安装ipykernel的所以无法注册到Jupyter Notebook中,所以先要准备下环境
  2. #安装ipykernel
  3. conda install ipykernel
  4. #写入环境
  5. python -m ipykernel install --name pytorch --display-name "Pytorch for Deeplearning"

教程中下一步是定制 Jupyter Notebook ,大概意思就是设置一个 Jupyter Notebook 的工作目录,理解成 eclipse 的 workspace 就行。

设置好后从 Anaconda Navigator 打开 Jupyter Notebook,

 

我们打开pytorch for deeplearning,输入一行命令 import torch 并执行

没有报错说明我们已经成功安装 pytorch,可以愉快的使用了

安装失败或者环境选错的同学,就会出现这样的报错:

为什么自己查询 pytorch 版本号还能成功打印,但在 jupyter 里引入 torch 会报错?那是因为你没有选择已经配置好pytorch的那个环境。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/运维做开发/article/detail/813559
推荐阅读
相关标签
  

闽ICP备14008679号