Skip to content
Off-the-shelf python package of pytorch with CUDA support for Mac OS
Branch: master
Clone or download
Latest commit 3989a30 Jun 12, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
usr_local_lib updatet v1.0.1 Mar 26, 2019
BuildInstractions-1.0.1.md add instructions for 1.0.1 Apr 3, 2019
BuildInstractions-1.0rc1.md updatet v1.0.1 Mar 26, 2019
LICENSE updatet v1.0.1 Mar 26, 2019
README.md add mxnet link Jun 12, 2019

README.md

Pytorch OSX Build

Unfortunately, the Pytorch team does not release binary package for Mac OS with CUDA support. This project provides off-the-shelf binary packages. Both Python 2.7 and 3.7 are supported now!

很不幸,Pytorch团队不发布 Mac OS CUDA版。本项目提供 Mac OS 上编译好、可直接安装的Pytorch CUDA版本。本项目同时支持Python 2.7 和 3.7 了!

Attenion

  • 【2018.11.30】Compiled with OpenMP support with a guide.
  • 【2018.07.31】Currently, Nvidia provides CUDA 9.2 driver for Mac OS High Sierra (10.13) only. Therefore, do NOT download the CUDA 9.2 versions if you are using Sierra (10.12).

Releases

FileName pytorch CUDA CUDNN Compute Capability Compilation Time
torch-1.0.1-cp27-cp27m-macosx_10_13_intel.whl 1.0 10.0 7.4 3.5,5.2,6.1,7.0 2018-03-26
torch-1.0.1-cp37-cp37m-macosx_10_13_x86_64.whl 1.0 10.0 7.4 3.5,5.2,6.1,7.0 2018-03-26
torch-1.0-cp27-cp27m-macosx_10_12_intel.whl 1.0 10.0 7.4 3.0,3.5,5.2,6.1 2018-12-21
torch-1.0-cp37-cp37m-macosx_10_12_x86_64.whl 1.0 10.0 7.4 3.0,3.5,5.2,6.1 2018-12-21
torch-1.0-cp27-cp27m-macosx_10_12_intel.whl 1.0 9.0 7 3.0,3.5,5.2,6.1 2018-12-13
torch-1.0-cp37-cp37m-macosx_10_12_x86_64.whl 1.0 9.0 7 3.0,3.5,5.2,6.1 2018-12-13
torch-1.0.0a0+ff608a9-cp27-cp27m-macosx_10_13_intel.whl 1.0rc1 9.0 7 3.0,3.5,5.2,6.1 2018-11-30
torch-1.0.0a0+ff608a9-cp37-cp37m-macosx_10_13_x86_64.whl 1.0rc1 9.0 7 3.0,3.5,5.2,6.1 2018-11-30
torch-0.4.1-cp36-cp36m-macosx_10_12_x86_64.whl 0.4.1 9.0 7 3.0,3.5,5.2,6.1 2018-08-01
torch-0.4.1-cp27-cp27m-macosx_10_12_intel.whl 0.4.1 9.0 7 3.0,3.5,5.2,6.1 2018-08-01
torch-0.4.1-cp36-cp36m-macosx_10_12_x86_64.whl 0.4.1 9.2 7.1 3.0,3.5,5.2,6.1 2018-07-30
torch-0.4.1-cp27-cp27m-macosx_10_12_intel.whl 0.4.1 9.2 7.1 3.0,3.5,5.2,6.1 2018-07-30
...

You can find releases in release page.

你可以在Release页面找到发布版本。

Installation for Python 2.7

First, ensure your CUDA driver and cudnn is installed properly, and copy dependencies in folder usr_local_lib to path /usr/local/lib. Also, install OpenMP using Homebrew.

首先,确保CUDA驱动和cudnn正确安装,并且将文件夹usr_local_lib中的依赖项复制到路径/usr/local/lib。也要通过Homebrew安装OpenMP。

sudo mkdir /usr/local
sudo mkdir /usr/local/lib
sudo cp usr_local_lib/* /usr/local/lib/
brew install libomp
brew link --overwrite libomp

Second, uninstall the previous pytorch installtion by

其次,卸载之前版本的pytorch:

pip uninstall torch

Install the wheel package from this project:

安装:

pip install torch*.whl

Install torchvision:

安装torchvision:

pip install -U torchvision

Installation for Python 3

Install Python 3.x from Homebrew first, and then simply follow the guide for Python 2.7 and replace pip command with pip3 and python with python3.

首先从Homebrew安装Python 3.x,然后按照Python 2.7的安装步骤执行,注意将pip替换为pip3,并用python3启动python

Enjoy!

开始使用Pytorch吧!

Source Code

Source code from: https://github.com/pytorch/pytorch

Related Links

If you need Tensorflow builds for osx, go to this page: https://github.com/TomHeaven/tensorflow-osx-build

If you need MxNet builds for osx, go to this page: https://github.com/TomHeaven/mxnet_osx_build

如果你需要Tensorflow包,请看这个页面:https://github.com/TomHeaven/tensorflow-osx-build

如果你需要MxNet包,请看这个页面:https://github.com/TomHeaven/mxnet_osx_build

You can’t perform that action at this time.