# 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