Mac 安装tensorflowpytorch

mac2025-12-28  11

因为我已经装了anaconda navigator,所以,可以界面安装,就界面安装。tensorflow可以界面安装,注意的是安装的应该是2.0的版本,所以写法上面和前面的版本有所不同,测试代码可以看官网的代码。如下

import tensorflow as tf

mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data() x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([   tf.keras.layers.Flatten(input_shape=(28, 28)),   tf.keras.layers.Dense(128, activation='relu'),   tf.keras.layers.Dropout(0.2),   tf.keras.layers.Dense(10, activation='softmax') ])

model.compile(optimizer='adam',               loss='sparse_categorical_crossentropy',               metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)

model.evaluate(x_test,  y_test, verbose=2)

不行的话

https://www.jianshu.com/p/eaee1fadc1e9

上述博客的博主清楚的讲解了anaconda管理包或者环境。

pytorch没搜到,所以我是在终端安装的,可以看官网,来找合适自己的pytorch版本https://pytorch.org

首先进入你的虚拟环境,因为我的虚拟环境是这个名字,请用你自己的环境名代替python37

source activate python37 

之后,你终端前面括号里会变成(python37)而不是(base),从下图1变成下图2

然后在输入你在官网匹配的适合你的pytorch代码,下面的是符合我的代码。

conda install pytorch torchvision -c pytorch

这个期间你可能会遇到Segmentation Fault问题,或者An HTTP error occurred when trying to retrieve this URL. HTTP errors are often intermittent, and a simple retry will get you on your way.(我遇到的)

别理他们,重新运行conda install pytorch torchvision -c pytorch

直到

因为我之前后两个下载成功,前面的失败了。

这样你就完事了,在这个环境中打开一个python的ide,测试一下你的torch

import torch

torch.FloatTensor([1,2,3])

嗯,这样就ok了。其实真的很简单,哎,心好累。

最新回复(0)