PyTorch环境配置及安装

环境配置


舒适提示:为了更好的教程体验,提供视频、阅读地址python

Youtube: https://www.youtube.com/playlist?list=PLgAyVnrNJ96CqYdjZ8v9YjQvCBcK5PZ-Vgit

BiliBili: https://www.bilibili.com/video/av74281036/github

Github: https://github.com/xiaotudui/PyTorch-Tutorial机器学习

相关下载:https://pan.baidu.com/s/16koDDDvAFO7rNoi-uR-A_g 提取码:pahiide

在学习中,遇到任何问题,能够关注公众号:土堆碎念,直接在对话框中留言,我会不定时解答的~函数

公众号:土堆碎念,回复:pytorch,便可得到全部的数据集及安装包, 省去找软件、数据集的时间。(同步更新中,请勿急躁)工具


0. 前言

在计算机领域学习,最烦的一件事,就是安装软件,环境配置。并且,偏偏不如意的是,它仍是你学习进程的第一关。第一关就出现大Boss。这样,超级容易打击学习的动力。学习

因此,在我看来,入门的话,必定要快,要不求甚解,先把握总体。因此,个人教程安排顺序一些方法,不一样于其余千篇一概的教程,其中也许有本身的想法。ui

若是你在学习中,有什么建议,或者遇到什么问题,能够关注公众号:土堆碎念,直接留言,私信我。spa

1. Anaconda 下载

在机器学习,深度学习中,要用到大量的 package(就是各类工具包)。若是说,函数是一个工具,那么 package 就是一个工具包。一个个安装 package 很麻烦,并且容易出现疏漏。因而,就有了 Anaconda,这是一个集成了经常使用于科学分析(机器学习,深度学习)的大量package。

也就是说,你只要安装了 Anaconda,就安装了不少咱们以后要用的许多packages。(他还有不少功能,对入门帮助超大,后面再说)

Anaconda 的下载地址:这儿(目前,页面上是 Python3.7 的版本,我用的是3.6版本,推荐3.6版本

以前使用过3.7版本,也许会遇到一些问题。

推荐: Anaconda历史版本连接:这儿,选择这个版本: Anaconda 3-5.2.0

2. Anaconda 安装

双击进行安装,须要注意如下几点:

(1)记住安装路径,以后会用到

(2)跳过安装 Microsoft VSCode

为了检验是否安装成功,在开始菜单出,左击 Anaconda Prompt

若是能够成功打开,且左边有 (base),即安装成功。

3. 显卡配置(无 Nvidia 显卡的略过)

只要你打开任务管理器,在 GPU 那里看到了你的 NVIDIA 显卡,便可。说明你的硬件驱动,已安装。

若是你的 NVIDIA 显卡未显示,可使用驱动人生,或者去官网下载驱动。

4. 有序的管理环境

也许,你以后会遇到不一样的项目,须要使用到不一样版本的环境。好比这个项目要用到 pytorch 0.4,另外一个项目要用到 pytorch 1.0,若是你卸载了0.4版本,安装了1.0版本。那么下一次,你再碰到0.4版本,你就须要卸载1.0版本,安装0.4版本。很折腾。

Anaconda 集成的 conda 包就可以解决这个问题。它能够创造出两个屋子,相互隔离。一个屋子放 0.4 版本,一个屋子放 1.0 版本。你须要哪一个版本,就进哪一个屋子工做。

咱们首先使用 conda 指令建立一个屋子,叫作 pytorch。

指令以下:

conda create -n pytorch python=3.6

conda 是指调用 conda 包,create 是建立的意思,-n 是指后面的名字是屋子的名字, pytorch是屋子的名字(能够更改为本身喜欢的),python=3.6 是指建立的屋子,是 python3.6 版本。

以后,弹出提示,输入 y,便可安装。

安装成功后,输入如下指令:

conda info --envs

便可看到 conda 环境中,有新建的 pytorch 环境,右边的 * 号表示,当前你处于哪一个环境。

接下来,咱们要在 pytorch 环境中,安装 PyTorch,(有点绕),使用以下指令,进入 pytorch 环境。

conda activate pytorch

你能够看到左边的 base 变成了 pytorch,表明成功进入 pytorch 环境。

5. 安装 PyTorch 准备

终于到重头戏了,安装 PyTorch了。激动的打开官网,下拉到下面的页面。

(不得不说,PyTorch 在这方面作的真的好,不须要再人工安装 CUDA、cuDNN 之类的,所有都给你解决了。真的良心~之前安装真的繁琐。(吐槽一句,之前超麻烦的)

PyTorch Build 选择 Stable;选择系统;Package,Windows下推荐 conda,Linux 下推荐 pip;Python版本按照Anaconda的版本选择,我这里选择3.6,CUDA 推荐9.2。若是没有显卡的话,选择 None。个人选择如图所示:

复制下面的代码,以后,在开始菜单中,打开 Anaconda Prompt,查看最左边括号中是 base 仍是 pytorch。

若是是 base,使用 conda activate pytorch 进入 pytorch 环境中。以后粘贴便可。

果断输入y,以后就是漫长的等待。或者去休息吧,慢慢等着。看看剧啥的,等着它慢慢下吧。若是你的并不慢,恭喜恭喜。

6. 加速(可选)

有的时候,下载速度太慢了,能够利用我已经下载好的文件。

连接:https://pan.baidu.com/s/1cyEjHdluc4ufCkvtbgziqA 
提取码:bzfx 
复制这段内容后打开百度网盘手机App,操做更方便哦

将其中的 pytorch-1.3.0-py3.6_cuda92_cudnn7_0.tar 和 cudatoolkit-9.2-0.tar两个文件,复制到刚才 Anaconda 安装的目录下的 pkgs 文件夹中。

这两个文件,只能安装 pytorch1.3.0 和 cuda 9.2,仅限Windows。

以后,在 Anaconda Prompt 命令行窗口,多按几回 Ctrl + C 结束程序。而后,从新复制 PyTorch 安装命令,粘贴到命令行上,进行安装。

7. 验证是否安装成功

(1)在命令行左边为 pytorch 环境中,输入 python

(2)以后,输入 import torch,若是没有报错,意味着 PyTorch 已经顺利安装了

(3)接下来,输入 torch.cuda.is_available,若是是 True,意味着你可使用 GPU,若是是 False,意味着只能使用CPU。


往期推荐

学会这三个排版原则,你就能超过80%人的排版

你想过将你的人生游戏化吗?

用游戏的思路激励本身

寻找有趣或更有效率的事、工具和教程

相关文章
相关标签/搜索