基於Windows10環境下PyTorch1.3.1的安裝

一、安裝環境：

• 系統：WIN10_X64;
• 顯卡：GTX 1070Ti

二、版本信息：

• 顯卡驅動：441.20
• CUDA版本：10.2
• CUDNN版本：7.6.5

     

• PyTorch: 1.3.1

三、安裝 Anaconda3 和 PyCharm2018:

       方法：Anaconda 和 PyCharm 的安裝與配置

四、更新 Nvidia顯卡驅動：

（1）、去 Nvidia 官網查找與電腦顯卡相匹配的顯卡驅動，並下載；

            鏈接：https://www.nvidia.cn/Download/index.aspx?lang=cn

（2）、安裝驅動；

 

五、安裝 CUDA;

(1)、打開 Nvidia 控制面板，點擊系統信息，確定 CUDA 的版本信息；

（2）、去英偉達官網，下載合適辦事版本的 CUDA;

            鏈接：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

（3）、安裝 CUDA;

（4）、查看電腦系統環境中的系統變量；

（5）、驗證 CUDA 是否安裝成功；

nvcc -V

結果：

 

六、安裝 CUDNN;

鏈接：https://developer.nvidia.com/rdp/cudnn-download;

（1）、解壓 CUDNN;

（2）、複製 CUDNN 中的 lib、include、bin 至 C:\Program Files\NVIDIA\NVIDIA GPU Computing Toolkit\CUDA\v10.2

（3）、查看電腦系統環境中的系統變量；

七、安裝 PyTorch;

（1）、使用 Anaconda prompt 創建環境；

conda create -n pytorch_gpu pip python=3.7

（2）、進入創建好的環境裏；

conda activate pytorch_gpu

(3)、開始安裝 PyTotch1.31

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

conda install pytorch torchvision cudatoolkit=10.1

八、驗證 PyTorch 安裝是否成功：

（1）、打印隨機生成張量矩陣；

import torch
x = torch.rand(5,5)
print(x)

輸出如下：

tensor([[0.7078, 0.1424, 0.3411, 0.3987, 0.3476],
    [0.7534, 0.7137, 0.3489, 0.4226, 0.3640],
    [0.4104, 0.8411, 0.5112, 0.0629, 0.0664],
    [0.7568, 0.9495, 0.3300, 0.2392, 0.6441],
    [0.7615, 0.1883, 0.6001, 0.9663, 0.3313]])

如下圖：

（2）、 驗證 GPU 是否可以用

torch.cuda.is_available()

輸出如下：

True

如下圖：

---------------------------------------

OK! PyTorch 安裝成功