Pytorch的安装教程从0开始

初始安装环境准备:

笔者初始安装环境:

win10虚拟机(物理机已经成功安装)


配置初始环境

首先需要安装conda环境,方便后期的配置和管理

安装conda

首先访问清华大学的镜像站点,读者也可以自行到官网下载,由于conda官网在外网的缘故,访问和下载的速度都比较慢(不使用梯子的情况下)下段附上官网地址和镜像站点地址
清华大学镜像站点conda下载直通车icon-default.png?t=N4P3https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/


conda官网下载icon-default.png?t=N4P3https://www.anaconda.com/download/

 笔者将以镜像站点为例做下载演示

  1. 找到符合自己的系统和版本

这里笔者选择的是Anaconda-2023.3的win64位版本,单击下载(需要耐心等待)

安装conda

这里笔者将在win64虚拟机中演示安装

这里选择安装路径

 随后便是一路next安装

 点击Next完成安装

这个时候点击“win”键打开任务栏,发现新安装的conda,右键点击,打开文件所在位置

 打开所在文件夹后将所有或者需要的快捷方式都拖到桌面


配置虚拟环境

点击打开Anaconda Prompt程序,这会打开一个类似CMD命令行的窗口,接下来的一些步骤都要在其中进行

创建新的虚拟环境

输入

conda create -n env_name python=x.x

其中env_name和python=x.x都是可改选项,前者(env_name)是你的自定义虚拟环境的名字

后者(python=x.x)其中x.x是python的版本,笔者这里使用的是3.10版本

接下来需要输入y以确认下载一系列安装包(请确保网络环境稳定

下载过程请耐心等待

 下载完毕后输入代码即可进入虚拟环境(此处activate后的虚拟环境名称是笔者的环境名称,注意识别)

conda activate pytorch

 成功进入虚拟环境之后路径前缀应该是虚拟环境的名称而非初始的(base)如图

 接下来将在此虚拟环境中安装pytorch


安装pytorch

首先访问pytorch的官网

PyTorch官网下载命令直通车icon-default.png?t=N4P3https://pytorch.org/get-started/locally/

 选择合适的安装,这里选择的意思是Windows系统conda环境python编程语言CPU支持,最下面的是运行命令

这里笔者选择的是比较稳的CPU支持安装,读者可以自行去根据电脑的型号去选择CUDA版本

复制提供的命令,图中的命令则是

conda install pytorch torchvision torchaudio cpuonly -c pytorch

接下来就是安装重灾区,直接原因便是安装的速度快慢问题,笔者这里提供一种比较稳的方法(因人而异,但是笔者觉得逻辑上来看这是比较稳的方法了)

首先在虚拟环境中运行这行命令,下载相关的附属安装包,等待一整子后关闭窗口重新打开,再次进入虚拟环境,输入下载命令,查看是否附属安装包都下载完毕,下图可以看见只有torch一个包正在下载了

 这个时候关闭窗口,进入pytorch的whl安装包的界面

whl文件下载直通车icon-default.png?t=N4P3http://download.pytorch.org/whl/torch/这里选择合适的whl文件(根据上图中的版本寻找对应版本进行选择和安装)

 这里笔者选择的是torch-2.0.1+cpu-cp310-cp310-win_amd64的版本,这里读者可以自行选择

选择成功后点击下载即可

随后再次进入虚拟环境,输入

pip install 文件路径

就像这样

 然后回车安装即可(耐心等待,如果遇到下载超时的问题,重新下载即可)

等待片刻后下载完成

验证下载成功

下载完毕后如何验证成功呢,先进入下图的软件,这是一个可视化的模块,在这里可以检查虚拟环境以及其下载的包

 可以看见torch模块已经成功安装在环境中了

 还可以在命令行中输入

python

进入python环境

import torch
torch.__version__

如果正确安装应该显示torch的版本,如笔者下图

 结语

安装教程就到这里结束了,不过保险起见可以再装上numpy模块,使用pip安装即可

希望这篇文章能对您有帮助,如果觉得文章不错,可以点赞收藏加关注,如果觉得还有地方错误或者欠佳,可以私信作者,小生必当斧正!

感谢浏览

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
当然没问题!下面我将为您提供一份简单的教程,手把手教您如何使用PyTorch。 首先,确保您已经安装PyTorch。您可以从官方网站(https://pytorch.org)根据您的操作系统和需求下载并安装PyTorch。 接下来,让我们从一个简单的示例开始。假设我们要训练一个简单的线性回归模型来预测房价。我们将使用均方差损失作为我们的损失函数,并使用随机梯度下降作为优化器。 首先,导入所需的库: ```python import torch import torch.nn as nn import torch.optim as optim ``` 然后,定义我们的模型。在这个例子中,我们将使用一个简单的线性层作为我们的模型。 ```python class LinearRegression(nn.Module): def __init__(self, input_size, output_size): super(LinearRegression, self).__init__() self.linear = nn.Linear(input_size, output_size) def forward(self, x): return self.linear(x) ``` 接下来,定义一些训练数据和标签。在这个例子中,我们将使用一个简单的2维张量。 ```python # 训练数据 x_train = torch.tensor([[1], [2], [3], [4]], dtype=torch.float32) # 标签 y_train = torch.tensor([[2], [4], [6], [8]], dtype=torch.float32) ``` 现在,我们可以实例化我们的模型并定义损失函数和优化器。 ```python # 实例化模型 model = LinearRegression(input_size=1, output_size=1) # 定义损失函数和优化器 criterion = nn.MSELoss() optimizer = optim.SGD(model.parameters(), lr=0.01) ``` 接下来,我们可以开始训练我们的模型了。训练过程通常包括多个迭代,每个迭代中都会进行前向传播、计算损失、反向传播和更新参数的步骤。 ```python # 设置训练迭代次数 num_epochs = 100 # 开始训练 for epoch in range(num_epochs): # 前向传播 outputs = model(x_train) # 计算损失 loss = criterion(outputs, y_train) # 反向传播和优化 optimizer.zero_grad() loss.backward() optimizer.step() # 打印训练进程 if (epoch+1) % 10 == 0: print('Epoch [{}/{}], Loss: {:.4f}'.format(epoch+1, num_epochs, loss.item())) ``` 最后,我们可以使用训练好的模型来进行预测。 ```python # 预测 x_test = torch.tensor([[5]], dtype=torch.float32) predicted = model(x_test) print('预测值: {:.2f}'.format(predicted.item())) ``` 这就是一个简单的使用PyTorch进行线性回归的示例。您可以根据您的需求扩展和修改这个示例,以适应不同的问题和模型。 希望这个教程对您有所帮助!如果您有任何问题,请随时问我。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值