安裝pytorch
環境:win10、pytorch1.6.0、cuda10.1、cudnn7.6.5
安裝步驟(打開 anaconda powershell prompt):
cuda與cndnn的安裝網絡上有很多,在此不做贅述
- 添加清華鏡像源
依次輸入以下指令:
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
- 建立虛拟環境
建立:
conda create --name pytorch_env python=3.6
激活:
conda activate pytorch_env
- 安裝pytorch (耐心等待)
conda install pytorch torchvision cudatoolkit=10.1
- 檢查是否安裝成功(本人在此處傳回了False)
輸入python進入指令行
>>> import torch
>>> torch.cuda.is_available()
若傳回True則成功,若傳回False則失敗
解決torch.cuda.is_available()為False的問題
問題:電腦安裝的顯示卡驅動低于CUDA版本所需
- 檢視電腦的顯示卡驅動版本
- win+r,輸入“cmd”打開“指令行提示符”視窗
- 輸入:
檢視系統的顯示卡驅動資訊nvidia-smi
- 發現
低于所安裝的CUDA Version: 10.0
版本cuda10.1
- 更新系統的顯示卡驅動
- 下載下傳驅動精靈
- 打開驅動管理,找到
-
- 點選更新,下載下傳完成之後,一步步執行安裝即可
- 重新在指令行輸入
,發現已更新為nvidia-smi
CUDA Version: 10.2
- 再次檢查是否安裝成功,發現
傳回為Truetorch.cuda.is_available()
注:除了更新系統驅動,也可以選擇降低cuda的安裝版本