pytorch学习(一)

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())


测试gpu是否可用

2.PyTorch 实现模型训练的 5 方面

(1)数据:包括数据读取,数据清洗,进行数据划分和数据预处理,比如读取图片如何预处理及数据增强。

(2)模型:包括构建模型模块,组织复杂网络,初始化网络参数,定义网络层。

(3)损失函数:包括创建损失函数,设置损失函数超参数,根据不同任务选择合适的损失函数。

(4)优化器:包括根据梯度使用某种优化器更新参数,管理模型参数,管理多个参数组实现不同学习率,调整学习率。

  (5)迭代训练:组织上面 4 个模块进行反复训练。包括观察训练效果,绘制 Loss/ Accuracy 曲线,用 TensorBoard 进行可视化分析。

3.pycharm安装与使用

1.安装一路点击社区版即可(免费)

2.创建项目以后,点击file,如图所示找到设置


pycharm使用anaconda环境
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容