PyTorch简介与安装-PyTorch

概述

安装anaconda

登录anaconda的官网下载,anaconda是一个集成的工具软件不需要我们再次下载。
anaconda官网
PyTorch简介与安装-PyTorch_第1张图片
点击Get Started进行下载页面。
PyTorch简介与安装-PyTorch_第2张图片
跳转到这个页面如果你的Python版本正好是3.8版,那便可以直接根据系统去选择自己相应的下载版本就可以了。
PyTorch简介与安装-PyTorch_第3张图片
但是如果你的Python版本号不是当前页面的版本号,那我建议你去选择相对应的版本号。
PyTorch简介与安装-PyTorch_第4张图片
点击archive你就会跳转到下面的页面。
下载完便开始安装,根据你的需求去选择,看你是商业用途还是个人用途,我选择的是个人。
PyTorch简介与安装-PyTorch_第5张图片
一定要记住安装路径。
PyTorch简介与安装-PyTorch_第6张图片
点击对勾将anaconda的默认值设置为python3.7
PyTorch简介与安装-PyTorch_第7张图片
PyTorch简介与安装-PyTorch_第8张图片
剩下的默认就可以了。
安装完成以后点击win图片,在最近添加和anaconda包中都能找到anaconda prompt。
PyTorch简介与安装-PyTorch_第9张图片
出现(base)便表示成功。
PyTorch简介与安装-PyTorch_第10张图片

环境管理

在我们做项目的时候可以能需要不同环境的python版本,有时候要1.0版本,有的需要3.0版本的拿在这里我们就需要建立不同的环境,在不同的需要的时候去使用。
这个是在Anaconda Prompt下操作的
在这里插入图片描述
这是创建的格式
"pytorch"是这个环境变量的名字
"python=3.7"是我们要确定的当前环境的版本数

我们输入activate pytorch(pytorch是你定义的这个环境的名字),左边的环境就从base(基本环境),变成了pytorch环境。
PyTorch简介与安装-PyTorch_第11张图片
输入pip list,查看当前环境下面有哪些包,我们发现没有pytorch,那么下面我们就需要安装它。
PyTorch简介与安装-PyTorch_第12张图片

PyTorch安装

进入PyTorch官网
PyTorch简介与安装-PyTorch_第13张图片
PyTorch简介与安装-PyTorch_第14张图片
先复制这一段操作指令。(这种情况是需要电脑上有单独的英伟达的显卡、或者英伟达的显卡和集显这两种情况都是可以的。)
下面我们看下电脑的nvidia版本。
PyTorch简介与安装-PyTorch_第15张图片
进入命令符号窗口,输入nvidia-smi,查看当前驱动的版本号,观察Driver Version的值是否大于400,如果小于请更新显卡驱动。推荐使用驱动精灵更新,虽然驱动精灵的版本没有官网的更新的那么及时但是驱动精灵更新简单,不需要过多的操作。当然也可以去英伟达显卡驱动更新去下载。
PyTorch简介与安装-PyTorch_第16张图片
选择自己显卡相对应的系列,括号中Notebooks是笔记本。

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

将上边复制的代码粘贴进去。
PyTorch简介与安装-PyTorch_第17张图片
输入y
在这里插入图片描述
便开始下载了(下载时间较长请耐心等待)
PyTorch简介与安装-PyTorch_第18张图片
下载完成后我们再次输入pip list,查看到已经有torch的存在。

检验安装

在这里插入图片描述
先输入python,然后输入import torch,如果输入后没有任何报错,没有任何显示那就是成功了,然后再输入torch.cuda.is_available(),返回的是True,那便是完成了整个操作。

安装下载过慢解决方案

简介

清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
清华源地址可以加快下载速度。

解决方案

conda中添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

另外为了保险起见,建议同时添加第三方conda源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

复制上面pytorch的安装地址,注意去除 -c pytorch,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包。

conda install pytorch torchvision torchaudio cudatoolkit=10.2

接着等待安装成功就好了。
最后可以通过命令行验证是否安装成功。
PyTorch简介与安装-PyTorch_第19张图片

参考

安装PyTorch详细过程
conda安装Pytorch下载过慢解决办法(11月26日更新ubuntu下pytorch1.3安装方法)
Pytorch环境配置与安装

你可能感兴趣的:(人工智能,python,人工智能)