一文详解Pytorch环境搭建:Mac电脑pip安装Pytorch开发环境
对于希望在本地环境中进行深度学习开发的开发者来说,配置PyTorch工具是至关重要的一步。。对于Mac用户而言,搭建PyTorch开发环境并不复杂,本文将详细介绍如何在Mac电脑上使用pip安装PyTorch开发环境,帮助开发者快速上手深度学习开发。
一、准备工作
在开始之前,请确保您的Mac系统中已安装以下软件,并确保Python版本至少为3.9版本
:
- Python:建议安装较高版本的Python3。
- pip:Python的包管理器,用于安装OpenCV库。
可以在终端,输入以下命令检查Python和pip的版本:
// 查看python版本
python3 --version
// 查看pip3版本
pip3 --version
在终端输入which python3 查看安装位置:
// 查看python安装位置
which python3
// 查看pip3安装位置
which pip3
如果您的Python3版本较低(低于3.9
),请先升级Python3版本。
二、安装Python
访问 PyTorch官方网站,根据您的 Mac 系统参数选择适合的PyTorch版本。在页面最下方,可以看到pip的安装链接。
PyTorch官方下载地址:
https://pytorch.org/get-started/locally/
安装前,请确认已安装python3和pip3。然后在终端直接输入pip3 install torch torchvision torchaudio
命令,进行torch安装。
三、验证Python
创建torch_demo.py
文件,写入如下两行代码:
# 导入PyTorch库
import torch
# 创建一个5行3列的随机张量(tensor)
x = torch.rand(5, 3)
# 打印随机张量(tensor)
print(x)
终端命令行,执行python3 torch_demo.py
运行程序,若正常输出版本信息,则tensor
相关信息,则表明安装成功。