MAC OS本地安装PyTorch

Categories 机器学习
MAC OS本地安装PyTorch

前言

PyTorchs是基于Python的机器学习操作库,他可以利用GPU的资源来进行复杂的深度学习运算。

pytorch

如果需要充分利用Pytorch的CUDA支持,需要电脑上有NVIDIA GPU。不过本人电脑是Macbook Air,没有这样的条件。且入门学习实验,希望使用CPU支持即可。

官网上说明,Mac OS如果要使用CUDA支持,需要源代码编译PyTorch:

Currently, CUDA support on macOS is only available by building PyTorch from source

安装前准备

macOS 版本

Mac OS 10.10 (Yosemite) or 更高

Python

Mac OS is 默认安装的Python 2.7. 一般建议使用Python 3.6或更高版本,这样与PyTorch搭配更好(官方称Python 2.7 也可使用)。Python 3.6可以使用Anaconda 或者Homebrew包管理器安装,或者从Python官网安装 

Anaconda包管理器的安装

打开Mac OS的终端,执行如下命令即可:

$curl -O https://repo.anaconda.com/archive/Anaconda3-5.2.0-MacOSX-x86_64.sh
$sudo sh Anaconda3-5.2.0-MacOSX-x86_64.sh

安装后,可以在终端输入如下命令验证:

$which python
/Users/xh/anaconda3/bin/python

$ python –version
Python 3.6.5 :: Anaconda, Inc.

安装PyTorch

使用Anaconda安装PyTorch也非常简单:

conda install pytorch torchvision -c pytorch

验证PyTorch安装

可将如下代码保存为test.py文件,执行python test.py进行验证

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

执行后输出类似于:

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]]

至此,安装结束。

猜你喜歡

神经网络和深度学习(1) 前言 机器学习,神经网络,已经不算前言的名词、未来的趋势。未来大部分的工作都可能会被机器所取代,这不是危言耸听。秉承着不能被机器淘汰的思想,必须了解到机器到底在学习什么,怎么能学习。本系列文算是个人对机器学习的一个笔记,加上自己的一些总结和变化。记录的过程即是加深学习的过程,希望可以今年内...
神经网络和深度学习(2) — 后向传播算法原理... 前言 神经网络和深度学习(1)中,我们从代码中看到机器学习使用了Back Propagation,但是并未介绍到其工作原理。本文则着重介绍Back Propagation算法(以下简称BP算法)的工作原理。本文大部分内容引用、整理或翻译自Michael Nielsen的《Neural Net...
Mithell《机器学习》学习笔记 – Chapter 1&2... 机器学习 绪论 应用 语音识别 自动车辆驾驶 西洋双陆棋 .... 相关学科 人工智能 贝叶斯方法 计算复杂性理论 信息论 控制论 统计学 问题的标准描述 任务T: 描述机器学习要做任务 性能标准P: 描述机器学习结果...

版权声明:本站文章除非特别声明,均为L&H原创。允许转载,转载时请务必以超链接形式标明文章 原始出处 、作者信息和本声明。

本文链接:MAC OS本地安装PyTorch @ L&H site


留个言吧大佬

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据