我正在学习PyTorch并遇到问题。如果我使用torch.xxx,尝试运行代码时总会出现AttributeError。
这是我的环境:
Windows 10,CUDA 10.0.132,Pycharm 2018.3.3,Python 3.7。
我使用了Anaconda提示符,并按照https://pytorch.org/get-started/locally/上的说明安装了PyTorch。这是命令:
点安装https://download.pytorch.org/whl/cu100/torch-1.0.0-cp37-cp37m-win_amd64.whl
pip install torchvision
然后在上述环境所在的路径中,将PyCharm中的解释器设置为python.exe。
以下是我在出现错误后尝试的方法:
将torch.xx更改为torch._C.xx
pip安装pylint
将解释器设置为系统解释器,而不是conda解释器
没有任何帮助。
这是最初的代码:
import torch
import matplotlib.pyplot as plt # 画图的模块
x = torch.unsqueeze(torch.linspace(-2, 2, 500), dim=1)
错误:
Traceback (most recent call last):
File "E:/Python/Winter/torch.py", line 1, in <module>
import torch
File "E:\Python\Winter\torch.py", line 6, in <module>
x = torch.unsqueeze(torch.linspace(-2, 2, 500), dim=1)
AttributeError: module 'torch' has no attribute 'unsqueeze'
然后我尝试了这个:
import torch
print(torch.FloatTensor([1, 2, 3], [4, 5, 6]))
相同错误:
Traceback (most recent call last):
File "E:/Python/Winter/torch.py", line 1, in <module>
import torch
File "E:\Python\Winter\torch.py", line 3, in <module>
print(torch.FloatTensor([1, 2, 3], [4, 5, 6]))
AttributeError: module 'torch' has no attribute 'FloatTensor'
我想正确安装和使用PyTorch。
谢谢。