Mac 安装tensorflow/pytorch

本文介绍了如何在Mac上通过Anaconda界面安装TensorFlow 2.0,并提供了安装过程中的测试代码。对于PyTorch,由于未找到界面安装方式,博主在终端中进行了安装,并详细说明了激活虚拟环境、查找合适版本及解决安装过程中遇到的问题的过程。
摘要由CSDN通过智能技术生成

因为我已经装了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
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值