這個原因是pytorch不支持CUDA,
可以先輸入
import torch
print(torch.cuda.is_available())
如果輸入為false則打開cmd,輸入nvidia-smi查看cuda的版本,之后去Previous PyTorch Versions | PyTorch找符合自己cuda的下載方式,重新下載一遍即可
這個原因是pytorch不支持CUDA,
可以先輸入
import torch
print(torch.cuda.is_available())
如果輸入為false則打開cmd,輸入nvidia-smi查看cuda的版本,之后去Previous PyTorch Versions | PyTorch找符合自己cuda的下載方式,重新下載一遍即可
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。