pytorch
ヾ云胡๑不喜ゞ
这个作者很懒,什么都没留下…
展开
-
epoch batch_size iterations
初学分不清这几个概念。以diabetes糖尿病数据集为例,数据集包含760条记录,八列数据epoch 表示周期,把这760条记录跑完一遍,是一个epochbatch_size 表示从数据集中抽取样本的数量,取2的n次方,iteration表示迭代多少次在一个epoch中,batch_size = 32,这里iteration=24,每个epoch中迭代24次...原创 2021-09-17 19:48:15 · 100 阅读 · 0 评论 -
安装GPU版本pytorch
第一步,首先安装CUDA先查看NVIDIA对应CUDA的版本直接打开NVIDIA控制面板,打开帮助,查看系统信息 ,点击组件,可以看到对应的CUDA版本是11.4,选择安装CUDA时,选择小于11.4的即可按照提示安装,在安装过程选择自定义安装,不要轻易修改路径,我在修改了CUDA安装路径后,cmd验证失败,默认失败,直接安装,cmd验证nvcc -V显示版本型号表示安装成功CUDA下载地址CUDA Toolkit Archive | NVIDIA DevelopercuD.原创 2021-08-26 22:13:46 · 116 阅读 · 0 评论