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 圖像分類示範的訓練時間。
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLicmbw5CZ1ITY3M2YxEWM3EWNxQTOkRWN0MjNyAjYmdzMwU2Yw8CX5d2bs92Yl1iclB3bsVmdlR2LcNWaw9CXt92Yu4GZjlGbh5yYjV3Lc9CX6MHc0RHaiojIsJye.png)
自己編譯
如果您發現版本不能滿足您的要求,您可以自己從源代碼編譯。
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