2021-08-05

cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

在跑Pytorch官方的Bilstm-CRF模型代码的时候,希望可以利用实验室的服务器的GPU加速训练。

第一次尝试

下面是我第一次尝试使用的部分代码

# GPU
import torch
dev_id = 1
dev = torch.device(f'cuda:{dev_id}')
# 将模型转入到GPU
model = model.to(dev)
# 将相应的数据和loss函数也要转移到GPU中
# sentence就是模型训练输入的数据,tags是对应的标签 
sentence = sentence.to(dev)
tags = tags.to(dev)

loss = model.neg_log_likelihood(sentence, tags)
loss.to(dev)

但是loss函数在计算的时候一直出现
cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

第二次尝试

将代码里面的.to(dev)全部改成.cuda(dev)就成功了,没有再出现
cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
目前我也不太清楚原因

# GPU
import torch
dev_id = 1
dev = torch.device(f'cuda:{dev_id}')
# 将模型转入到GPU
model = model.cuda(dev)
# 将相应的数据和loss函数也要转移到GPU中
# sentence就是模型训练输入的数据,tags是对应的标签 
sentence = sentence.cuda(dev)
tags = tags.cuda(dev)

loss = model.neg_log_likelihood(sentence, tags)
loss.cuda(dev)
这个数据可以使用Python进行解析和处理。可以按照以下步骤进行: 1. 将数据分割成每个数据项。 ``` data_items = data.split(',') ``` 2. 对于每个数据项,将其按#分割成四个小项,并存储到一个列表中。 ``` data_list = [] for item in data_items: item_list = item.split('#') data_list.append(item_list) ``` 3. 对于每个小项,进行相应的类型转换。 ``` for item in data_list: item[0] = int(item[0]) # 题号转换为整数 item[1] = datetime.datetime.strptime(item[1], '%Y-%m-%d %H:%M:%S') # 时间转换为datetime类型 if item[2] != '': # 操作类型转换为整数 item[2] = int(item[2]) item[3] = str(item[3]) # 科目转换为字符串类型 ``` 4. 可以按照需要对数据进行进一步处理,如按照题号、时间等进行排序、筛选等操作。 完整的Python代码如下: ``` import datetime data = '''1#2021-05-18 14:31:55##初级会计实务,2#2021-05-18 14:31:57#12#初级会计实务,2#2021-05-18 14:32:08##初级会计实务,3#2021-05-18 14:32:09#12#初级会计实务,4#2021-05-18 14:32:34#12#初级会计实务,4#2021-05-18 14:32:45##初级会计实务,5#2021-05-18 14:32:46#12#初级会计实务,5#2021-05-18 14:32:57##初级会计实务,6#2021-05-18 14:33:00#12#初级会计实务,7#2021-05-18 14:33:44#12#初级会计实务,7#2021-05-18 14:34:42##初级会计实务,8#2021-05-18 14:34:43#12''' # 将数据分割成每个数据项 data_items = data.split(',') # 对于每个数据项,将其按#分割成四个小项,并存储到一个列表中 data_list = [] for item in data_items: item_list = item.split('#') data_list.append(item_list) # 对于每个小项,进行相应的类型转换 for item in data_list: item[0] = int(item[0]) # 题号转换为整数 item[1] = datetime.datetime.strptime(item[1], '%Y-%m-%d %H:%M:%S') # 时间转换为datetime类型 if item[2] != '': # 操作类型转换为整数 item[2] = int(item[2]) item[3] = str(item[3]) # 科目转换为字符串类型 print(data_list) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值