天天看點

anaconda安裝torch_win10安裝pytorch并解決torch.cuda.is_available()為False

安裝pytorch

環境:win10、pytorch1.6.0、cuda10.1、cudnn7.6.5

安裝步驟(打開 anaconda powershell prompt):

cuda與cndnn的安裝網絡上有很多,在此不做贅述
  1. 添加清華鏡像源
依次輸入以下指令:
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
檢視是否成功添加:
conda config --show channels           
  1. 建立虛拟環境
建立:
conda create --name pytorch_env python=3.6
激活:
conda activate pytorch_env           
  1. 安裝pytorch (耐心等待)
conda install pytorch torchvision cudatoolkit=10.1           
  1. 檢查是否安裝成功(本人在此處傳回了False)
輸入python進入指令行
>>> import torch
>>> torch.cuda.is_available()
若傳回True則成功,若傳回False則失敗           

解決torch.cuda.is_available()為False的問題

問題:電腦安裝的顯示卡驅動低于CUDA版本所需
  • 檢視電腦的顯示卡驅動版本
  1. win+r,輸入“cmd”打開“指令行提示符”視窗
  2. 輸入:

    nvidia-smi

    檢視系統的顯示卡驅動資訊
  3. 發現

    CUDA Version: 10.0

    低于所安裝的

    cuda10.1

    版本
anaconda安裝torch_win10安裝pytorch并解決torch.cuda.is_available()為False
  • 更新系統的顯示卡驅動
  1. 下載下傳驅動精靈
  2. 打開驅動管理,找到
anaconda安裝torch_win10安裝pytorch并解決torch.cuda.is_available()為False
    1. 點選更新,下載下傳完成之後,一步步執行安裝即可
    2. 重新在指令行輸入

      nvidia-smi

      ,發現已更新為

      CUDA Version: 10.2

anaconda安裝torch_win10安裝pytorch并解決torch.cuda.is_available()為False
  • 再次檢查是否安裝成功,發現

    torch.cuda.is_available()

    傳回為True
注:除了更新系統驅動,也可以選擇降低cuda的安裝版本

繼續閱讀