mac m1 使用conda创建tensorflow环境,用gpu进行训练

解决的问题:m1使用gpu训练。或者可以直接用谷歌的colab进行,但是慢。

一、首先安装homebrew

  1. (推荐使用)因为按照官网安装出现错误,用下面大佬镜像很快
/bin/zsh -c "$(curl -fsSL https://gitee.com/cunkai/HomebrewCN/raw/master/Homebrew.sh)"
  1. homebrew官方安装

二、使用brew安装miniforge

brew install miniforge

miniforge会安装在homebrew文件夹下,接下来的安装也都会在mini forge下

conda install -y jupyter

三、安装tensorflow可能会使用到的库

随便在一个目录下新建如下文件

vim tensorflow-apple-metal.yml

然后复制下面内容到文,这个是依赖库,可以自行添加自己需要的。这样做的好处是可以为不同的任务创建不同的库。

name: tensorflow
 
dependencies:
    - python=3.9
    - numpy
    - pip>=19.0
    - jupyter
    - apple::tensorflow-deps
    - scikit-learn
    - scipy
    - pandas
    - pandas-datareader
    - matplotlib
    - pillow
    - tqdm
    - requests
    - h5py
    - pyyaml
    - flask
    - boto3
    - pip:
        - tensorflow-macos
        - tensorflow-metal
        - bayesian-optimization
        - gym
        - kaggle
        

使用:wq保存退出,接下来使用命令创建环境,需要在刚才保存tensorflow-apple-metal.yml这个文件下进行

conda env create -f tensorflow-apple-metal.yml -n tensorflow

四、安装其他

到上一步,已经新建好tensorflow文件,接下来进行简单安装,为了可以在jupterbook中使用,依次使用下面代码

在这一步可能会出现问题,无法激活环境,我的原因是powershell 中使用的是zsh,需要conda init zsh,这样就可以了。但是还有个问题就是打开terminal默认使用conda环境,还未学习如何处理。

conda activate tensorflow

conda install nb_conda

python -m ipykernel install --user --name tensorflow --display-name "Python 3.9 (tensorflow)"

然后打开jupter notebook,打开方式就是直接输入

jupyter notebook

测试代码

# What version of Python do you have?
import sys

import tensorflow.keras
import pandas as pd
import sklearn as sk
import tensorflow as tf

print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {tensorflow.keras.__version__}")
print()
print(f"Python {sys.version}")
print(f"Pandas {pd.__version__}")
print(f"Scikit-Learn {sk.__version__}")
gpu = len(tf.config.list_physical_devices('GPU'))>0
print("GPU is", "available" if gpu else "NOT AVAILABLE")

'''
Init Plugin
Init Graph Optimizer
Init Kernel
Tensor Flow Version: 2.5.0
Keras Version: 2.5.0

Python 3.9.6 | packaged by conda-forge | (default, Jul  6 2021, 08:51:19) 
[Clang 11.1.0 ]
Pandas 1.3.0
Scikit-Learn 0.24.2
GPU is available
'''

参考:https://github.com/jeffheaton/t81_558_deep_learning/blob/master/install/tensorflow-install-mac-metal-jul-2021.ipynb

你可能感兴趣的:(tensorflow,深度学习,神经网络)