在华为AI加速ECS上安装Pytorch和Tensorflow框架

#仅CPU

konda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch full, use list conda检查，如果安装了pytorch，则说明安装成功。

您还可以检查pytorch版本：

4。下载官方pytorch例程

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

使用git clone命令下载例程。 git 克隆 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

训练完成，模型文件存放在训练脚本目录中。 ？ 3. 验证安装

4。下载lenet例程

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

如果直接运行原代码，会报错：ImportError: unavailable to import name 'tutorial' from 'tensorflow_core 。例如'需要更改代码。

Python 训练 Train.py