win10 + Anaconda 安裝 pytorch-gpu【2020版 良心推薦】

win10 + Anaconda 安裝 pytorch-gpu

• 一、所需軟體
• • • • 1.Anaconda
      • 2.cuda 10.2
      • 3.cudnn v7.6.5
• 二、安裝cuda、cuDNN
• • • • 1.安裝cuda：直接安裝運作cuda，安裝過程時間有點長，要耐心等待哦QAQ
      • 2. 安裝cuDNN：
      • 3. 配置環境變量
• 三、安裝pytorch-gpu
• • • • 1.在Anaconda中建立一個pytorch的環境
      • 2.激活pytorch環境 ：
      • 3.在建立的pytorch環境中安裝pytorch-gpu
      • 4.測試是否成功安裝pytorch
• 四、将pytorch環境添加到jupyter notebook中
• • 方法一：簡單
  • 方法二：可以自由命名
  • • • 1.首先安裝ipykernel：
      • 2.在虛拟環境下建立kernel檔案：
      • 3.激活conda環境：
      • 4.将環境寫入notebook的kernel中：
      • 5.打開jupyter notebook
• 五、添加清華鏡像源
• • • • 1.添加清華鏡像源：
      • 2.檢視所有鏡像源：
      • 3.通過清華鏡像下載下傳pytorch：
• 六、番外

一、所需軟體

首先去pytorch官網檢視自己所需要下載下傳cuda、cudnn的版本

• cuda下載下傳官網

  

• cudnn下載下傳官網（需要新增賬號）

  

這裡我給出相應版本百度雲的連結，友善大家食用

1.Anaconda

連結：https://pan.baidu.com/s/1ZEFoYo8evR_7eySvvSUwTA

提取碼：q6qy

2.cuda 10.2

連結：https://pan.baidu.com/s/15BqEltYMd50mjXhEZQ4pQA

提取碼：jlwo

3.cudnn v7.6.5

連結：https://pan.baidu.com/s/1mGnATEDl0DJ06L7gErZ5SA

提取碼：0yni

二、安裝cuda、cuDNN

【安裝前提：具備Visual Studio的環境，不然按照精簡模式安裝會安裝失敗】

1.安裝cuda：直接安裝運作cuda，安裝過程時間有點長，要耐心等待哦QAQ

• 測試cuda是否安裝成功： 在終端輸入：

  nvcc -V

  

2. 安裝cuDNN：

• 将cuDNN裡面的庫檔案複制到cuda安裝目錄相應的檔案夾内，即解壓壓縮包，把壓縮包中bin,include,lib中的檔案分别拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2目錄下對應目錄中

  

• 把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64\cupti64_102.dll

  拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin

3. 配置環境變量

確定已經将上面4個path添加到系統變量中！

三、安裝pytorch-gpu

1.在Anaconda中建立一個pytorch的環境

在Anaconda Prompt 終端輸入：

conda create -n pytorch

2.激活pytorch環境 ：

在Anaconda Prompt 終端輸入：

conda activate pytorch

3.在建立的pytorch環境中安裝pytorch-gpu

在Anaconda Prompt 終端輸入從pytorch官網獲得的安裝指令：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

==============================================

**（網不穩定的小夥伴可以通過添加鏡像源哦，參見步驟五）**

==============================================

4.測試是否成功安裝pytorch

四、将pytorch環境添加到jupyter notebook中

方法一：簡單

在Anaconda Prompt 終端依次輸入：

activate pytorch

conda install nb_conda_kernels

方法二：可以自由命名

1.首先安裝ipykernel：

conda install ipykernel

2.在虛拟環境下建立kernel檔案：

conda install -n pytorch ipykernel

3.激活conda環境：

activate pytorch

4.将環境寫入notebook的kernel中：

python -m ipykernel install --user --name pytorch --display-name "pytorch"

删除指定kernel：

jupyter kernelspec remove pytorch

檢視所有kernel：

jupyter kernelspec list

5.打開jupyter notebook

五、添加清華鏡像源

1.添加清華鏡像源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
           

删除鏡像：

conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

2.檢視所有鏡像源：

conda config --show

3.通過清華鏡像下載下傳pytorch：

conda install pytorch torchvision cudatoolkit=10.2

=======================================

至此，大功告成！！！

=======================================

六、番外

在這裡向大家推薦兩篇不錯的文章，比我寫的更加清楚

windows10安裝pytorch，感覺自己基本踩了所有坑，這裡提供最有效的兩種方法

Import Error：Jupyter Notebook 中無法運作虛拟環境（pytorch、tensorflow）