一、下載M芯片的anaconda,并安裝
二 、安裝GPU版本的pytorch
1.安裝 Xcode
xcode-select --install
2.創(chuàng)建環(huán)境
conda create -n torch-gpu python=3.9
conda activate torch-gpu
3.打開pytorch官網(wǎng)復(fù)制命令,?注意:在mac m上,device是’mps’ 而不是’cuda’, mac的MPS支持MacOS 12.3+
?4.測(cè)試
import torch
import math
print(torch.backends.mps.is_available())
# True
print(torch.backends.mps.is_built())
# True
三、安裝GPU版本的tensorflow
1.安裝 Xcode
xcode-select --install
2.創(chuàng)建環(huán)境
conda create -n tensorflow-gpu python=3.9
conda activate tensorflow-gpu
3. 安裝conda install -c apple tensorflow-deps
conda install -c apple tensorflow-deps
?4.Install base TensorFlow
python -m pip install tensorflow-macos==2.9
5.??Install tensorflow-metal plug-in
python -m pip install tensorflow-metal==0.6
?6.驗(yàn)證
import tensorflow as tf
print(tf.test.is_gpu_available())
7.完成,附上我安裝完pytorch和tensorflow的圖文章來源:http://www.zghlxwxcb.cn/news/detail-536946.html
文章來源地址http://www.zghlxwxcb.cn/news/detail-536946.html
到了這里,關(guān)于蘋果mac m1,m2芯片安裝 pytorch和tensorflow的GPU版本的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!