PyTorch中的Cublas运行时错误？代码在Jupyter笔记本中工作，但在终端中不工作

Question

我是python新手，很难找出错误消息的来源。我在Jupyter笔记本中编写了一个代码来使用一些现有的（更复杂的）代码，它似乎工作得很好。我在Jupyter笔记本中收到了这个运行时错误消息，但一切似乎都很正常，我正在编写的文件都被创建了。我一直在尝试调整它，使其在anaconda3终端内部运行，现在一切都崩溃了。我创建了python文件，并将jupyter笔记本中相同的“工作代码”放入其中，现在运行时错误阻止代码运行。我不知道为什么它可以在jupyter笔记本中使用，但不能在终端中使用。 无论如何，这里是运行时错误“

RuntimeError: cublas runtime error : the GPU program failed to execute at C:/w/1/s/tmp_conda_3.7_055306/conda/conda-bld/pytorch_1556690124416/work/aten/src/THC/THCBlas.cu:450

我一直在谷歌搜索，有人建议更新CUDA。我试过了，但我认为它不起作用。我想知道是否有人知道这个错误的含义以及我如何解决它？ 这是我做nvcc --version时得到的有关CUDA版本的信息

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_May__6_19:10:02_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.167
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28358933_0

这是我在运行时错误之前得到的回溯：

  File "AMASSBody.py", line 42, in <module>
    body = bm(pose_body=pose_body, betas=betas, root_orient=root_orient)
  File "C:\Users\emmay\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "c:\users\emmay\src\human-body-prior\human_body_prior\body_model\body_model.py", line 251, in forward
    dtype=self.dtype)
  File "C:\Users\emmay\anaconda3\lib\site-packages\smplx\lbs.py", line 179, in lbs
    v_shaped = v_template + blend_shapes(betas, shapedirs)
  File "C:\Users\emmay\anaconda3\lib\site-packages\smplx\lbs.py", line 265, in blend_shapes
    blend_shape = torch.einsum('bl,mkl->bmk', [betas, shape_disps])
  File "C:\Users\emmay\anaconda3\lib\site-packages\torch\functional.py", line 211, in einsum
    return torch._C._VariableFunctions.einsum(equation, operands)

我正在windows 10上运行python3。如果你们需要更多信息，请告诉我