問題介紹

在安裝torch之后，命令行（Anaconda Powershell Prompt）運(yùn)行這三行代碼：

python # python
import torch
torch.cuda.is_available()

返回結(jié)果始終為False。

出錯(cuò)原因

原因有多個(gè)，可以參考文章最后的鏈接【1】

他的很清晰，如果按我的沒有解決可以看一下。

主要就是以下兩個(gè)：

1. CUDA、cudnn、torch版本不對應(yīng)。（解決方法參考鏈接【1】)
2. 一個(gè)坑：是通過清華源下載的！
   檢查是不是清華源下載導(dǎo)致的問題：

conda list

看到Pytorch的Build是pyxxx_cpu_0，就說明下的pytorch是cpu版本的。這樣就會導(dǎo)致輸出False。

導(dǎo)致的原因可能是清華源沒有對應(yīng)cuda版本的pytorch，所以會裝cpu版的。參考鏈接【2】

我的問題就是通過清華源下載導(dǎo)致的。下面是解決方案

解決方案

上官網(wǎng)下載：https://pytorch.org/get-started/previous-versions/
找到你的CUDA版本支持的pytorch（我的是CUDA10.1）

然后復(fù)制對應(yīng)的pip語句

——重點(diǎn)！復(fù)制pip語句，而不是下圖這種conda語句，因?yàn)槟阋呀?jīng)切換了清華源

然后在命令行（或Anaconda Prompt中運(yùn)行）

記得不要科學(xué)上網(wǎng)，否則可能會報(bào)錯(cuò) ValueError: check_hostname requires server_hostname

等待，安裝就可以啦

要是下載速度很慢，可以參考鏈接【1】。

參考內(nèi)容：

【1】torch.cuda.is_available()返回false——解決辦法

【2】完美解決torch.cuda.is_available()一直返回False的玄學(xué)方法

總結(jié)