PyTorch报”NameError: name ‘Variable’ is not defined “的原因以及解决办法

目录

• 更改Variable为torch.Tensor：
• 确认PyTorch版本是否正确：
• 总结

在PyTorch中，Variable已经被torch.Tensor取代了。因此，导致“NameError: name 'Variable' is not defined”的问题通常是由于使用了已经被弃用的Variable而不是新的torch.Tensor。

解决该问题的方法有以下几种：

更改Variable为torch.Tensor：

import torch

# define a tensor
x = torch.Tensor([1, 2, 3, 4])

# define a variable (deprecated)
#v = Variable(x)

# print the tensor
print(x)

确认是否导入了torch.autograd.Variable：

如果确认使用的是新版本的PyTorch，但仍出现了“NameError: name 'Variable' is not defined”的问题，可能是由于没有导入torch.autograd.Variable。在代码开头加上以下语句：

from torch.autograd import Variable

确认PyTorch版本是否正确：

如果以上两种方法均不起作用，则可能是由于PyTorch版本不正确。请确保使用的是最新版本的PyTorch。可以使用以下命令安装最新版本的PyTorch

pip install -U torch torchvision

总结

在PyTorch中使用torch.Tensor代替Variable，并确认导入了正确的库，即可解决“NameError: name 'Variable' is not defined”的问题。