在pycharm中模型訓(xùn)練時(shí)，出現(xiàn)錯(cuò)誤提示：AssertionError: Torch not compiled with CUDA enabled
原因：CUDA和Torch版本不兼容
1.嘗試檢查cuda版本
使用命令nvcc -V或者 nvcc --version
出現(xiàn)如下所示結(jié)果，則表明cuda安裝成功：
否則就說明沒有cuda，要重新下載安裝cuda，去官網(wǎng)：
https://developer.nvidia.com/cuda-downloads
檢查電腦適配cuda版本：打開cmd，輸入nvidia-smi，結(jié)果顯示：
選擇更多之前版本：
根據(jù)自己電腦的系統(tǒng)選擇合適的進(jìn)行下載即可：
安裝包下載好后，右鍵——管理員運(yùn)行——進(jìn)入安裝，無腦點(diǎn)下一步即可（盡量按照系統(tǒng)為你選擇好的安裝路徑來安，也就是C盤路徑，需要注意：在“安裝選項(xiàng)”這一欄選擇：自定義安裝。
如果你的電腦中有CUDA但版本不合適，刪除重裝即可，需要注意的是，一定要卸載干凈。我是在電腦自帶的控制面板中進(jìn)行卸載的：點(diǎn)擊控制面板——程序——程序和功能。找到所有和NVIDIA有關(guān)的全部卸載，然后再去軟件管家清理卸載殘留。其他的卸載方式也可，但一定要卸載干凈。
安裝完成后，檢查是否安裝成功，再次打開cmd，輸入
nvcc -V
2.pytorch環(huán)境的配置
首先對(duì)照CUDA和pyTorch對(duì)應(yīng)的版本：
選擇好版本后我們就開始下載pytorch安裝包
方法有兩種：
1.在pytorch官網(wǎng)：https://pytorch.org/get-started/locally/選擇在線安裝，接著訪問 Pytorch 的官網(wǎng)，找到與你本機(jī) CUDA 對(duì)應(yīng)版本的 Pytorch，并進(jìn)行安裝，具體操作如下：
若你的 CUDA 版本沒有出現(xiàn)在官網(wǎng)給出可選的選項(xiàng)范圍內(nèi)，則進(jìn)入下載其他版本的入口進(jìn)行歷史版本的下載：
2.離線安裝
參考文章：https://blog.csdn.net/weixin_72056722/article/details/126651703?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126651703-blog-125737765.pc_relevant_multi_platform_whitelistv3&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1-126651703-blog-125737765.pc_relevant_multi_platform_whitelistv3&utm_relevant_index=1
以上安裝完成后，可以進(jìn)行簡單的版本檢查測(cè)試：
>>> improt torch
>>> print(torch.__version__)
>>> print(torch.cuda.is_available())
最后輸出的結(jié)果如果是：
1.11.0+cu113
True
恭喜你，安裝成功！
但如果是像我這樣的倒霉蛋??：可能是這樣的：
建議直接檢查顯卡型號(hào)：
在設(shè)備管理器中，win+x選擇“設(shè)備管理器”，找到“顯示適配器”，
如果出現(xiàn)上面的樣子，建議換個(gè)機(jī)子或者外接顯卡或外接服務(wù)器了(* ^ _ ^ *)
如果有NVIDIA的顯卡話，可以嘗試重新安裝一下顯卡驅(qū)動(dòng)：
參考文章：https://blog.csdn.net/qq_45507678/article/details/124130123
NVIDIA官網(wǎng)：https://www.nvidia.cn/geforce/drivers/
第二個(gè)選項(xiàng)欄目是所有的型號(hào)系列
例如1050顯卡
臺(tái)式電腦選擇GeForce 10 Series，筆記本電腦選擇GeForce 10 Series（Notebooks）
還有一種倒霉蛋，電腦是Macbook的，就別想了，蘋果好像和英偉達(dá)鬧掰了，不支持，外接吧，或者服務(wù)器，別問我為什么知道(* ^ _ ^ *)
cuda的卸載重裝：文章來源：http://www.zghlxwxcb.cn/news/detail-784364.html
https://blog.csdn.net/weixin_44606139/article/details/127493438文章來源地址http://www.zghlxwxcb.cn/news/detail-784364.html
到了這里，關(guān)于python安裝cuda出現(xiàn)錯(cuò)誤提示：AssertionError: Torch not compiled with CUDA enabled的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
