出现错误:AssertionError: If capturable=False, state_steps should not be CUDA tensors
原因:pytorch 1.12.0的bug,官方1.12.1版本做出修复
解决方法:删除重装
安装命令如下:
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -c pytorch
亲测有效
出现错误:AssertionError: If capturable=False, state_steps should not be CUDA tensors
原因:pytorch 1.12.0的bug,官方1.12.1版本做出修复
解决方法:删除重装
安装命令如下:
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -c pytorch
亲测有效