Pytorch:GPU版代碼改成CPU版(RuntimeError: torch.cuda.FloatTensor is not enabled.)

1. 問題:運行代碼時出現錯誤:RuntimeError: torch.cuda.FloatTensor is not enabled. 主要是因爲我電腦沒有英偉達顯卡,不支持GPU加速,所以安裝的PyTorch是cpu版本的,不是gpu版本的,不支持cuda。   但是,這個代碼作者說是此代碼時專爲用cuda運行而設計的,所以此處出錯了。 2. 解決:把cpu版本的代碼改成cpu版本的。在代碼裏
相關文章
相關標籤/搜索