本来代码里有一句device = ('cuda'if torch.cuda.is_available else 'cpu')
即用CUDA计算,否则用CPU。
但是一直报错AssertionError Torch not compiled with CUDA enabled。
查了半天也查不到确切的问题,一怒之下把代码改成了device=('cpu')
好了。
贴上觉得有用的其他帖子。
https://blog.csdn.net/weixin_41243949/article/details/103363233
https://blog.csdn.net/qq_45504119/article/details/105405151
https://blog.csdn.net/m0_63483436/article/details/124122014
(为啥要说我文章质量低!?)