-
Notifications
You must be signed in to change notification settings - Fork 281
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
error #10
Comments
也是这个报错 |
似乎是CUDA错误? |
python3.8也出现这个问题
|
pip3 install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html |
我用Python3.8成功安装了你推荐的这个,可还是出现相同的报错。 |
我的环境为:1.7.0+cu101+python3.7.9,该环境没问题, |
项目文件夹cmd打开 |
cpu下torch不支持半精度,所以half() 出现的地方需要修改为float。 |
@sssssshf 感谢!直接将AIDetector_pytorch.py文件里面的half()全部替换为float()就可以了 |
RuntimeError: "unfolded2d_copy" not implemented for 'Half'
The text was updated successfully, but these errors were encountered: