1.注意事项:
(1)安装pytorch一定注意cuda版本与安装版本匹配,cuda版本过高pytorch官网可能没有与之匹配的解决方法大致如下三种(Win10)
第一:去官网换成版本低的cuda,建议备份现在版本。最简单
第二:下载另外一个版本在另一个盘,然后更改path路径。
第三:GitHub有编译pytoch的可以尝试一下。
(2)cuda下载pytorch东西很多,可以用pip指令下载。
(3)检查gpu是否可用代码如下
import torch
print(torch.cuda.is_available());
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())
2.PyTorch 实现模型训练的 5 方面
(1)数据:包括数据读取,数据清洗,进行数据划分和数据预处理,比如读取图片如何预处理及数据增强。
(2)模型:包括构建模型模块,组织复杂网络,初始化网络参数,定义网络层。
(3)损失函数:包括创建损失函数,设置损失函数超参数,根据不同任务选择合适的损失函数。
(4)优化器:包括根据梯度使用某种优化器更新参数,管理模型参数,管理多个参数组实现不同学习率,调整学习率。
(5)迭代训练:组织上面 4 个模块进行反复训练。包括观察训练效果,绘制 Loss/ Accuracy 曲线,用 TensorBoard 进行可视化分析。
3.pycharm安装与使用
1.安装一路点击社区版即可(免费)
2.创建项目以后,点击file,如图所示找到设置