一、前期準備
1. Anaconda 安裝
歷史版本anaconda下載,推薦下載3-5.2.0版本,該版本對應python3.6。
檢測anaconda是否安裝成功:
在開始菜單出,左擊 Anaconda Prompt (如果開始菜單中沒有出現此選項,一定安裝失敗)如果可以成功打開,且左邊有 (base),即安裝成功。
2.顯卡配置(需要保證有Nvidia 顯卡 版本在960M及以上)
首先查詢顯卡驅動版本:打開命令行輸入nvidia-smi
查看版本信息,需要保證大於396.26
若小於396.26,應先更新驅動。推薦使用驅動精靈 更新顯卡驅動,更加方便快捷
3.使用conda創建新的Python環境 (可選)
推薦創建新的Python環境將PyTorch安裝在其中
左擊 Anaconda Prompt 打開,輸入以下指令創建環境:
conda create -n torchenv python=3.6
conda 是指調用 conda 包,create 是創建的意思,-n 是指後面的名字是環境的名字, pytorch是環境的名字(可以更改成自己喜歡的),python=3.6 是指創建的屋子,是 python3.6 版本。
之後,彈出提示,輸入 y,即可安裝。
安裝成功後,輸入以下指令將當前環境置爲新創建的環境:
conda activate torchenv
二、PyTorch安裝
推薦使用清華源進行安裝,優點安裝速度快、成功率高,但版本爲舊版本(一樣用)
接着上面步驟
在 Anaconda Prompt 中,輸入以下命令,修改源
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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
最後,輸入以下命令進行安裝:
conda install pytorch torchvision cudatoolkit=10.0
大概等待2h左右安裝成功,具體視網絡情況而定 。
三、檢測是否安裝成功
接上一步操作,在Anaconda Prompt中
輸入 python,打開python環境
輸入 import torch ,如果沒有報錯,意味着 PyTorch 已經順利安裝了
輸入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。
【參考】
[1] PyTorch環境配置及安裝
[2] pytorch快速安裝【清華源】