conda中pycuda import pycuda.gpuarray报错

在anaconda中安装pycuda，版本2019.1.2：

运行demo的时候出现：

import pycuda.gpuarray as gpuarray  #导入GPU端的数组
  File "D:\zahid\Anaconda3\envs\pt-gpu\lib\site-packages\pycuda\gpuarray.py", line 4, in <module>
    import pycuda.elementwise as elementwise
  File "D:\zahid\Anaconda3\envs\pt-gpu\lib\site-packages\pycuda\elementwise.py", line 38, in <module>
    from pytools import memoize_method
  File "D:\zahid\Anaconda3\envs\pt-gpu\lib\site-packages\pytools\__init__.py", line 249
    __slots__: List[str] = []
             ^
SyntaxError: invalid syntax

 

根据错误提示应该和pytools有关系，查看版本后，pytools为2020.3，然后尝试更换版本。先卸载原来的版本，然后安装2019.1版本后，问题消失。