2024/10/25

install pytorch with mcp support

開啟 ssh server 後就可以用 ssh 登入。
下載 Miniconda3-latest-MacOSX-arm64.sh, 執行。
最後一樣會修改 .zshrc
一樣刪掉,放到 condaenv.sh

一樣, 但是default python 好像是 3.13 太新。torchvision 安裝會有問題。
所以要指定
conda create -n pytorch python==3.11
完成後,進入 pytorch env, follow apple support的說明 (要用 nightly channel):
conda install pytorch torchvision torchaudio -c pytorch-nightly
測試:
>> import torch
>> torch.backends.mps.is_available()
True
所以有 support metal GPU

另外,mps 是 metal GPU
Neural Engine 是 pytorch-lightning

沒有留言:

張貼留言