Anaconda環境中PyTorch GPU版本安裝(2020.03最新+避坑+全面)

一、前期準備

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快速安裝【清華源】

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章