首先設(shè)置conda的清華源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/# 設(shè)置搜索時顯示通道地址 conda config --set show_channel_urls yes

然后在官網(wǎng)查找對應(yīng)的版本，如

若要使用清華源安裝，則需去掉后面的-c pytorch，即

conda install pytorch torchvision torchaudio cudatoolkit=10.2

安裝后，import torch，再 torch.cuda.is_available()，若為true，則可使用GPU

在這里，我顯示false

可見NVIDIA的驅(qū)動有些老舊，所以升級N卡驅(qū)動

在右下角右鍵英偉達(dá)標(biāo)志，并進(jìn)入GeForce experience界面，下載驅(qū)動并安裝

安裝過程中顯示geforce game ready driver無法繼續(xù)安裝，可能是未刪除上一版本的驅(qū)動，但我們也不必手動刪除

進(jìn)入C:\ProgramData\NVIDIA Corporation\Downloader，選擇剛剛下載好的驅(qū)動

雙擊進(jìn)行安裝

注意選擇清潔安裝，安裝后可以看到驅(qū)動變化

?

總結(jié)

以上是生活随笔為你收集整理的pytorch GPU版安装 与N卡显卡驱动升级 踩坑的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。