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

安装好 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 环境中,输入:

import torch
torch.__version__

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

下面我们还需要配置 Jupyter Notebook

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

#新建的环境是没有安装安装ipykernel的所以无法注册到Jupyter Notebook中,所以先要准备下环境
#安装ipykernel
conda install ipykernel
#写入环境
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的那个环境。

  • 10
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值