我试图测量torch中一些代码的时间,但无法示例化torch.cuda.Event
。documentation不建议需要任何参数,并且同一行在不同的机器上工作。
卸载并重新安装torch(在Windows 10上使用pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
)并没有解决这个问题。
下面是我得到的错误:
Python 3.9.5 (tags/v3.9.5:0a7dcbd, May 3 2021, 17:27:52) [MSC v.1928 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.Event(enable_timing=True)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Users\john\AppData\Local\Programs\Python\Python39\lib\site-packages\torch\cuda\streams.py", line 138, in __new__
return super(Event, cls).__new__(
TypeError: object.__new__() takes exactly one argument (the type to instantiate)
字符串
我会感谢任何帮助在这里!
2条答案
按热度按时间htrmnn0y1#
我在不同的机器上多次遇到这个问题。每次,程序都运行在
cpu
上,而不是gpu
上。确保它在gpu
上运行为我解决了这个问题。我没有设法找到一个解决方案,使它在
cpu
上工作vybvopom2#
这些是我第一眼想到的可能的问题,请检查一下
torch
安装torch.cuda.is_available()
GPU
是否支持您的CUDA
版本?