参考:
Pytorch 报错 AssertionError: If capturable=False, state_steps should not be CUDA tensors.-CSDN博客
解决方案:
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
参考:
Pytorch 报错 AssertionError: If capturable=False, state_steps should not be CUDA tensors.-CSDN博客
解决方案:
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116