解决Torch not compiled with CUDA enabled问题

实际上是torch版本问题,说明当前的Pytorch版本无法使用显卡

先卸载原先pytorch版本

pip uninstall torch
pip uninstall torchvision

然后下载torch 1.9.1版本以及torchvision 0.10.1版本

conda install torch=1.9.1
conda install torchvision=0.10.1


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部