首先設置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/# 設置搜索時顯示通道地址 conda config --set show_channel_urls yes

然后在官網查找對應的版本，如

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

conda install pytorch torchvision torchaudio cudatoolkit=10.2

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

在這里，我顯示false

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

在右下角右鍵英偉達標志，并進入GeForce experience界面，下載驅動并安裝

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

進入C:\ProgramData\NVIDIA Corporation\Downloader，選擇剛剛下載好的驅動

雙擊進行安裝

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

?

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。