Mac下安装PyTorch

1. 查看Python版本

Mac上默认装有Python 2.7，先看一下python的版本：

在这里插入图片描述

目前推荐使Python 3，有关python的安装教程请参见python安装。

安Python 3时会默认安装pip3。输入命令pip3 -V或者pip3 --version可以查看pip3的版本信息。

注意：记得更新pip源使得快速安装各种安装包。

#创建.pip文件夹
$ mkdir ~/.pip
#创建pip.conf文件
$ touch touch ~/.pip/pip.conf
#打开pip.conf文件
$ vim ~/.pip/pip.conf
#填入以下内容更换为清华源
[global]
index-url=https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=pypi.tuna.tsinghua.edu.cn

2. 虚拟环境

推荐使用virtualenv安装。virtualenv 是一个和其它Python项目开发隔离的虚拟Python环境，在同一台机器上不会干扰也不会被其它程序影响。总而言之，virtualenv提供了一个安全可靠的PyTorch安装和运行机制。

#安装virtualenv
$ sudo pip3 install -U virtualenv

#创建一个新的虚拟环境，选择Python解释器并创建一个./venv目录来存放安装包
$ virtualenv --system-site-packages -p python3 ./venv
#可以根据偏好修改虚拟环境目录名称

#激活该虚拟环境
$ source ./venv/bin/activate
#当virtualenv处于有效状态时，shell提示符带有(venv)前缀

#升级pip
(venv) $ pip install --upgrade pip

#查看虚拟环境内的所有安装包
(venv) $ pip list

之后可以使用以下命令退出virtualenv：

# 在使用完PyTorch之前不要退出
(venv) $ deactivate

3. 安装PyTorch

#安装PyTorch CPU最新版本
(venv) $ pip install torch torchvision

运行以下代码验证安装效果：

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]])