Pytorch学习之环境的安装
各位小伙伴,博主也是刚开始学习Pytorch。在安装过程中也是看了好多帖子,下面给大家一个详细的过程,希望能帮助同样是小白的你。
首先,你要确认一下你的电脑有没有NVIDIA显卡,如果没有的话,你的Pytorch是不支持GPU加速的,只能进行简单的学习。下面先介绍一下如何查看你的电脑是不是含有NVIDIA显卡,首先在桌面上“此电脑”这个图标进行右键然后选择管理这个选项,接下来就按照下图去查看就可以了。
以上就是前期的准备工作,下面我就开始教大家如何去安装Pytorch
1.安装Anacode
如果你去官网下载的话,那速度可是比乌龟还慢,所以我就直接介绍快速下载的方法,我们需要去清华源镜像网站去下载,地址如下清华源镜像中心
点进去以后按下图进行操作就可以了
接下来你会发现一个神奇的事情,几分钟我们就把Anacode给下载完了,然后就是一直下一步进行安装就可以了。安装完以后找到Anacode打开。按下图所示去打开终端
接下里就是要安装Pytorch了。
2.安装Pytorch
在Ancode里打开终端界面以后在里面输入以下的命令:
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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.2
此时,你会发现,那安装的速度是飞快飞快的,几百M的文件一会就安装好了,比官网安装的速度不知快了多少倍。
因为博主是安装的最新版本,所以最后一行语句是等于10.2,
3.下面就是检查我们的Pytorch是不是安装好了
首先打开cmd窗口,先切换到python环境中,然后依次输入下面三个语句
import torch
import tprchvision
print(torch.cuda.is_availabl())
如果最后的结果显示True,那么恭喜你,你已经成功的安装好了Pytorch。
4.下面说一下显示False的原因
检查你的电脑是不是有NVIDIA显卡
如果有的话,还显示False那就是你的显卡驱动版本太低,你直接去NVIDIA官网上更新到最新的驱动就好了。
如果你的电脑没有NVIDIA显卡的话,而且你的import torch和import torchvision这两个语句也没有报错的话,就说明你的Pytorch已经安装好了,你可以在cmd窗口下的python环境中运行以下的语句去测试一下:
import torch
x = torch.empty(5,3)
print(x)
如果显示下图也就说明你的Pytorch已经安装好了