ASSERTION ERROR: Torch未启用CUDA支持的解决方法

在運行pytorch代碼的過程中，突然出現錯誤提示

AssertionError: Torch not compiled with CUDA enabled

這表明

1. 你確實已經完成了pytoch的安裝過程

2. 你所安裝的版本為僅支持cpu運算的類型

為了進一步確認情況，可以嘗試在python編輯器中輸入以下代碼

解决方案

首先，需完成Nvidia toolkit的安装流程

     sudo apt install nvidia-cuda-toolkit
    
    

接下来，需将pytorch进行卸载操作，具体步骤可参考下图所示。

    conda uninstall pytorch
    

若未采取相应措施，其可能仅对不同版本的cpu进行更新操作，具体示例如下截图所示。

接下来需完成pytoc