求助新分配账号无法正常使用gpu。爆显存
发表于 : 2023-04-13 18:23
入门小白被折磨几天了....
ubuntu20.04服务器新分配的账号ssh远程连接,深度学习入门阶段。
装好环境后无法正常使用GPU,一开始使用MMdetection开源项目的demo推理一张图片,模型一加载就报错RuntimeError: CUDA error: out of memory
尝试改官方提供 的demo配置文件config也没用。同学提醒使用cpu推理正常,自己手搓一个超小的tensor送进GPU也是报错RuntimeError: CUDA error: out of memory。
ubuntu系统管理员分配新账号难道还需要设置给GPU使用权限吗?还是说GPU坏了?(nvidia-smi查看全部是0显存0占用率,可用显存11GB,无正在运行的程序,确定不是其他人占用了GPU)
测试GPU的python脚本:
import torch
from torch import nn
class mynet(nn.Module):
def __init__(self):
super().__init__()
self.fc = nn.Linear(2, 1, bias=False)
def forward(self, x):
return self.fc(x)
net = mynet()
input = torch.randn(1, 2)
print(net, input)
device = torch.device('cuda:0')
#device = torch.device('cpu')
#cpu 运行成功
net.to(device)
input.to(device)
print(net, input)
print(net(input))
ubuntu20.04服务器新分配的账号ssh远程连接,深度学习入门阶段。
装好环境后无法正常使用GPU,一开始使用MMdetection开源项目的demo推理一张图片,模型一加载就报错RuntimeError: CUDA error: out of memory
尝试改官方提供 的demo配置文件config也没用。同学提醒使用cpu推理正常,自己手搓一个超小的tensor送进GPU也是报错RuntimeError: CUDA error: out of memory。
ubuntu系统管理员分配新账号难道还需要设置给GPU使用权限吗?还是说GPU坏了?(nvidia-smi查看全部是0显存0占用率,可用显存11GB,无正在运行的程序,确定不是其他人占用了GPU)
测试GPU的python脚本:
import torch
from torch import nn
class mynet(nn.Module):
def __init__(self):
super().__init__()
self.fc = nn.Linear(2, 1, bias=False)
def forward(self, x):
return self.fc(x)
net = mynet()
input = torch.randn(1, 2)
print(net, input)
device = torch.device('cuda:0')
#device = torch.device('cpu')
#cpu 运行成功
net.to(device)
input.to(device)
print(net, input)
print(net(input))