【2023 · CANN训练营第一季】——在华为AI加速型ECS上安装Pytorch和Tensorflow框架

# CPU Only

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch

        安装完成后，用conda list查看，有pytorch表示安装成功。

        也可以查看pytorch版本：

4、下载pytorch官方例程

地址：https://github.com/pytorch/examples

使用git clone命令下载例程。git clone https://github.com/pytorch/examples.git

5、跑一个CPU训练

        选择mnist例程，进行cpu训练。

    （1）修改代码

        例程综合考虑了cpu、gpu、mac gpu下的训练，我们这里使用的是cpu训练，需要将gpu、mac gpu关闭，避免因为检查gpu、mac gpu而报错。

        用Visual Studio Code修改训练脚本如下：

        修改参数，将训练好的模型落盘存储

    （2）运行训练脚本   python main.py

        训练完成，模型文件存储到训练脚本目录下。

四、安装Tensorflow1.15

1、切换至base，创建tensorflow虚拟环境——conda activate base

2、安装tensorflow 1.15 cpu版本

            conda install tensorflow=1.15.0

3、安装验证

4、下载lenet例程

        git clone https://gitee.com/qmckw/lenet.git

        原始代码直接运行会报错：ImportError: cannot import name ‘tutorials’ from ‘tensorflow_core.examples’，需要改一下代码。

        执行训练 python Train.py