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已经安装好了
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

麦田里的梵先生

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值