pytorch `torch.cuda.Event()` throws error `TypeError:object.__new__()只接受一个参数(要示例化的类型)`

kxxlusnw  于 2023-08-05  发布在  其他
关注(0)|答案(2)|浏览(104)

我试图测量torch中一些代码的时间,但无法示例化torch.cuda.Eventdocumentation不建议需要任何参数,并且同一行在不同的机器上工作。
卸载并重新安装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)

字符串
我会感谢任何帮助在这里!

htrmnn0y

htrmnn0y1#

我在不同的机器上多次遇到这个问题。每次,程序都运行在cpu上,而不是gpu上。确保它在gpu上运行为我解决了这个问题。
我没有设法找到一个解决方案,使它在cpu上工作

vybvopom

vybvopom2#

这些是我第一眼想到的可能的问题,请检查一下

  • 检查您的torch安装torch.cuda.is_available()
  • 您的GPU是否支持您的CUDA版本?

相关问题