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

2023-05-3016:51:37服务器及运维Comments423 views字数 729阅读模式

# CPU Only文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

安装完成后,用conda list查看,有pytorch表示安装成功。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

也可以查看pytorch版本:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

4、下载pytorch官方例程文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

地址:https://github.com/pytorch/examples文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

使用git clone命令下载例程。git clone https://github.com/pytorch/examples.git文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

5、跑一个CPU训练文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

选择mnist例程,进行cpu训练。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

(1)修改代码文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

例程综合考虑了cpu、gpu、mac gpu下的训练,我们这里使用的是cpu训练,需要将gpu、mac gpu关闭,避免因为检查gpu、mac gpu而报错。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

用Visual Studio Code修改训练脚本如下:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

修改参数,将训练好的模型落盘存储文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

(2)运行训练脚本   python main.py文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

训练完成,模型文件存储到训练脚本目录下。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

四、安装Tensorflow1.15文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

1、切换至base,创建tensorflow虚拟环境——conda activate base文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

2、安装tensorflow 1.15 cpu版本文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

conda install tensorflow=1.15.0文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

3、安装验证文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

4、下载lenet例程文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

git clone https://gitee.com/qmckw/lenet.git文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

原始代码直接运行会报错:ImportError: cannot import name ‘tutorials’ from ‘tensorflow_core.examples’,需要改一下代码。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

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

执行训练 python Train.py文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html

华为AI加速型ECS上安装Pytorch和Tensorflow框架
文章源自菜鸟学院-https://www.cainiaoxueyuan.com/yunwei/43466.html
  • 本站内容整理自互联网,仅提供信息存储空间服务,以方便学习之用。如对文章、图片、字体等版权有疑问,请在下方留言,管理员看到后,将第一时间进行处理。
  • 转载请务必保留本文链接:https://www.cainiaoxueyuan.com/yunwei/43466.html

Comment

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定