Pytorch錯誤：Torch not compiled with CUDA enabled

先測試：

import torch

print(torch.__version__)

print(torch.cuda.is_available())

如果輸出的結(jié)果是False，那么說明當前的Pytorch版本無法使用顯卡。

解決辦法：

（1）如果有顯卡資源并需要使用顯卡資源，先檢查一下當前的CUDA版本，在終端中輸入

cat /usr/local/cuda/version.txt

（2）然后根據(jù)自己的CUDA版本在Pytorch官網(wǎng)查看并安裝可用的版本，參考鏈接如下:

CUDA 10.2

https://download.pytorch.org/whl/cu102/torch_stable.html

CUDA 10.1

https://download.pytorch.org/whl/cu101/torch_stable.html

CUDA 10.0

https://download.pytorch.org/whl/cu100/torch_stable.html

CUDA 9.0

https://download.pytorch.org/whl/cu90/torch_stable.html

CPU及所有CUDA版本

https://download.pytorch.org/whl/torch_stable.html

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容