2021年10月30日
RuntimeError: cuda runtime error (30) : unknown errorr
摘要： 原因： config.yml中的 DEVICE配置出了问题。原文件中默认：DEVICE: 1 # 0: CPU, 1: GPU这里的配置与本机CUDA的起了冲突，在CUDA安装目录下的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\e 阅读全文
posted @ 2021-10-30 22:11 云梦幽幽 阅读(776) 评论(0) 推荐(0)
线性回归模型 B站唐宇迪学习pytorch（GPU版）框架
摘要： 复现视频里的代码，方便后续查阅。 import numpy as np import torch torch.cuda.current_device() import torch.nn as nn # 构造输入数据x和其对应的标签y x_values=[i for i in range(11)] x 阅读全文
posted @ 2021-10-30 21:44 云梦幽幽 阅读(280) 评论(0) 推荐(0)