python常用机器学习框架（torch、paddlepaddle、MindSpore）的安装方法

最新推荐文章于 2024-10-02 09:07:54 发布

拓宽视野

最新推荐文章于 2024-10-02 09:07:54 发布

阅读量2k

点赞数 1

分类专栏：机器学习 python后端文章标签： python 机器学习 paddlepaddle

本文链接：https://blog.csdn.net/Areigninhell/article/details/121703981

版权

python后端同时被 2 个专栏收录

24 篇文章 0 订阅

订阅专栏

机器学习

6 篇文章 0 订阅

订阅专栏

安装 tensorflow

# cpu版
    pip install tensorflow==1.14.0  -i https://mirrors.aliyun.com/pypi/simple/
# gpu版
    pip install tensorflow-gpu==1.14.0  -i https://mirrors.aliyun.com/pypi/simple/
# 验证
    python -c "import tensorflow;print(tensorflow.__version__)"

安装 keras

pip install keras==2.2.4 -i https://mirrors.aliyun.com/pypi/simple/
    python -c "import keras;print(keras.__version__)"

安装 torch

torch==1.7.1
# cpu版本
    pip install --no-cache-dir \
            --timeout=100 \
            torch==1.7.1+cpu \
            torchvision==0.8.2+cpu \
            torchaudio==0.7.2 \
            -f https://download.pytorch.org/whl/torch_stable.html \
            -i https://mirrors.aliyun.com/pypi/simple/ \
# gpu版
    conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
# 验证
     python -c "import torch;print(torch.__version__)"

安装 caffe

#cpu版：
    conda install -c defaults caffe
# gpu版：
    conda install -c defaults caffe-gpu
# 验证
   python -c "import caffe;print(caffe.__version__)"

安装 paddlepaddle

# cpu版
    pip install paddlepaddle==2.0.1  -i https://mirror.baidu.com/pypi/simple
# gpu版
    python -m pip install paddlepaddle-gpu==2.0.1.post101 \
                -f https://paddlepaddle.org.cn/whl/mkl/stable.html \
                 -i https://mirror.baidu.com/pypi/simple
    python -c "import paddle;print(paddle.__version__)"

安装 MindSpore

# cpu
    mamba install mindspore-cpu=1.5.0 -c mindspore -c conda-forge
# gpu
    conda install mindspore-gpu=1.5.0 cudatoolkit=10.1 -c mindspore -c conda-forge
# 验证
    python -c "import mindspore;mindspore.run_check()"