尝试用GPU跑torch,
torch.cuda.is_available()显示为True,但是torch.zeros(1).cuda()时卡住
查阅发现,因为CUDA的版本问题,30系显卡不支持10.系列的CUDA,需要换用11.1系列,具体安装教程可见这位博主:
https://blog.csdn.net/zc19981224/article/details/119705267
换用11.系列的CUDA后,不再卡住,能正常运行了
尝试用GPU跑torch,
torch.cuda.is_available()显示为True,但是torch.zeros(1).cuda()时卡住
查阅发现,因为CUDA的版本问题,30系显卡不支持10.系列的CUDA,需要换用11.1系列,具体安装教程可见这位博主:
https://blog.csdn.net/zc19981224/article/details/119705267
换用11.系列的CUDA后,不再卡住,能正常运行了