天天看點

MacOSX安裝Pytorch-GPU版本

Pytorch OSX 建構

不幸的是,Pytorch 團隊沒有釋出支援 CUDA 的 Mac OS 二進制包。該項目提供現成的二進制包。Both Python 2.7 and 3.7 are supported now!

很不幸,Pytorch 團隊不釋出 Mac OS CUDA 版本。本項目提供 Mac OS 上編譯好、可直接安裝的 Pytorch CUDA 版本。本項目同時支援Python 2.7 和 3.7 了!

性能警告

【2020.02.18】我在 Mac OS X 10.13.6 和 Ubuntu 16.04 上用 CUDA 10.1 和 CUDNN 7.6.5 對 Pytorch 1.3.1 進行了基準測試,在 Mac OS 上的性能大約是在 Ubuntu 上的 2/3。此外,在 Mac OS 上更容易遇到“CUDA OUT OF MEMORY”錯誤,因為作業系統需要大量的 GPU 記憶體來顯示。請注意這種性能差異,如果您要處理大量資料,最好轉向 Ubuntu!

下表列出了 MNIST 圖像分類示範的訓練時間。

MacOSX安裝Pytorch-GPU版本
MacOSX安裝Pytorch-GPU版本

自己編譯

如果您發現版本不能滿足您的要求,您可以自己從源代碼編譯。

1.0.1.

1.0rc1.

源路徑在source_pathesmaster 分支的檔案夾中可用。

安裝 Python 2.7

首先,確定您的 CUDA 驅動程式和 cudnn 安裝正确,并将檔案夾中的依賴項複制usr_local_lib到 path /usr/local/lib。另外,使用 Homebrew 安裝 OpenMP。

首先,確定 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

二、解除安裝之前的pytorch安裝

即将,解除安裝之前版本的pytorch:

pip uninstall torch

從這個項目安裝wheel包:

安裝:

pip install torch*.whl

安裝 torchvision:

pip install -U torchvision

安裝 Python 3

安裝從自制的Python 3.x的第一,然後隻要按照用于Python 2.7導向和替換pip用指令pip3和python用python3。

首先從Homebrew安裝Python 3.x,然後按照Python 2.7的安裝步驟執行,注意将pip替換為pip3,并用python3啟動python。

享受!

開始使用Pytorch吧!

源代碼

源代碼來自:

https://github.com/pytorch/pytorch