在复现代码时遇到这个错误
查了很多资料发现是torch和cuda版本问题,问了kimi后运行安装版本就好了
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
发现之前安装的torch的cpu版本的,重新下载了改成了cuda版本的。
在复现代码时遇到这个错误
查了很多资料发现是torch和cuda版本问题,问了kimi后运行安装版本就好了
pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
发现之前安装的torch的cpu版本的,重新下载了改成了cuda版本的。