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

2023年 7月 10日 18.6k 0

# CPU Only

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

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

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

也可以查看pytorch版本:

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

4、下载pytorch官方例程

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

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

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

5、跑一个CPU训练

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

(1)修改代码

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

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

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

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

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

(2)运行训练脚本   python main.py

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

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

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

四、安装Tensorflow1.15

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

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

2、安装tensorflow 1.15 cpu版本

conda install tensorflow=1.15.0

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

3、安装验证

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

4、下载lenet例程

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

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

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

执行训练 python Train.py

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

相关文章

服务器端口转发,带你了解服务器端口转发
服务器开放端口,服务器开放端口的步骤
产品推荐:7月受欢迎AI容器镜像来了,有Qwen系列大模型镜像
如何使用 WinGet 下载 Microsoft Store 应用
百度搜索:蓝易云 – 熟悉ubuntu apt-get命令详解
百度搜索:蓝易云 – 域名解析成功但ping不通解决方案

发布评论