os.environ['CUDA

余槿&流年: 您好 可是我使用原本数据维度为[240,60,1]的股票价格数据，出来的数据还是[240,60,1] 我的网络代码如下： class LSTMNet(nn.Module): def __init__(self): super(LSTMNet, self).__init__() # 这里input_size是1是因为数据集为股票价格 其中特征数量只有1个 即为当前日期前60天的股票价格（就只有股票价格，所以是一个特征数量） self.lstm1 = nn.LSTM(input_size=1, hidden_size=80,dropout=0.2) # 这里输出[240,60,80] # 没太搞懂batch_first是啥意思 self.lstm2 = nn.LSTM(input_size=80, hidden_size=100,dropout=0.2) # 这里输出[240,60,100] self.linear1 = nn.Linear(100, 1) # 经过这里输出[240,60,1] def forward(self, x): print(x.shape) # 原始维度：[240,60,1] x, _ = self.lstm1(x) # 只保留最后一个时间步的输出，忽略hidden state print(x.shape) # 原始维度：[240,60,80] x, _ = self.lstm2(x) # 只保留最后一个时间步的输出，忽略hidden state print(x.shape) # 原始维度：[240,60,100] x = torch.squeeze(x, dim=0) # 去除维度为1的维度 print(x.shape) output = self.linear1(x) return output[:,-1,:]

相关知识

CUDA Environment Variables: CUDA
No CUDA runtime is found, using CUDA
pytorch单机多卡训练 logger日志记录和wandb可视化
环境部署中cuda对应的tensorflow
Pytorch使用cuda后，任务管理器GPU的利用率还是为0？
CUDA Occupancy Calculator占用率计算与显卡算力计算
[YOLOv7]基于YOLOv7的动物识别系统(源码＆部署教程)
训练GAN时遇到的一些问题（更新中……）
快来给你的宠物视频加个表情特效吧
PyTorch单机多卡训练（无废话）

网址: os.environ['CUDA https://www.mcbbbk.com/newsview930572.html