AssertionError: Torch not compiled with CUDA enabled
地址:https://mirror.sjtu.edu.cn/pytorch-wheels/cu121/?经查阅,这个问题是因为conda默认安装的是CPU版本的torch,应该使用GPU版本的。如果直接从资料库安装,很慢,建议下载后手工安装。我的是12.0,算比较新的,就不更新了。这种不带后缀的,就CPU版本的。命令:nvidia-smi。
原创
2024-01-23 00:02:37 ·
961 阅读 ·
0 评论