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

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

一、首先安装homebrew

/bin/zsh -c "$(curl -fsSL https://gitee.com/cunkai/HomebrewCN/raw/master/Homebrew.sh)"

二、使用brew安装miniforge

brew install miniforge

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

conda install -y jupyter

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

在任意目录中创建以下文件

[En]

Create the following files in any directory

vim tensorflow-apple-metal.yml

然后将以下内容复制到文本中,这是一个依赖库,您可以添加您需要的内容。这样做的好处是可以为不同的任务创建不同的库。

[En]

Then copy the following to the text, this is a dependent library, you can add what you need. The advantage of this is that different libraries can be created for different tasks.

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
'''

Original: https://blog.csdn.net/weixin_42054232/article/details/120043474
Author: weixin_42054232
Title: mac m1 使用conda创建tensorflow环境,用gpu进行训练

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/514650/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球