參考資料

Mac OS 在 PyTorch 啟用 GPU

  1. 因為我本身使用我重新基于Anaconda安装了一下GPU支持的Pytorch，你不確定是否有安裝Conda可以使用指令conda --version查看是否有安裝。如果成供輸出帶有本身有安裝，如果沒有請至Anaconda官網下載。
  2. （可選）如果想要分開環境，特別建立一個支援GPU的python env，可以使用以下指令
1
2
3
4
5
6
7
8
# 建立一個 torch-gpu 的環境 使用 python 3.10.9
conda create -n torch-gpu python=3.10.9
# 啟用該環境
conda activate torch-gpu
# 查看 所有環境 
conda env list 
# 查看目前的 python 版本與啟用的 env python 版本是否相同
python --version

  1. 根據 Pytorch 官網選擇對應的版本，複製安裝指令

  2. 您可以使用簡單的 Python 腳本驗證 mps 支援：

1
2
3
4
5
6
7
8
9
10
11
import torch
if torch.backends.mps.is_available():
    mps_device = torch.device("mps")
    x = torch.ones(1, device=mps_device)
    print (x)
else:
    print ("MPS device not found.")

# 或是
print(torch.backends.mps.is_available()) # True
print(torch.backends.mps.is_built()) # True

需要注意的是，如果在 mac os 上，要使用 mps 而非一般 coda:0，要這樣寫：device = torch.device("mps)