1.檢測cuda：win+R,輸入cmd進入命令行，輸入nvidia-smi檢查nvidia驅動版本（必須nvidia驅動才能跑gpu），輸入nvcc -V
，出現如圖所示信息，則安裝cuda成功：
2.在cuda的include目錄下找到了cudnn.h，則證明cudnn安裝成功（cudnn是一套組件，是對cuda圖形處理功能的補充）。
（注：cuda在系統中的默認安裝位置為 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA，並且一般只能選擇默認安裝位置）
3.此處使用的是anaconda安裝pytorch，因此在pycharm中配置環境時，依次選擇 file --> setting --> project --> 點擊
Project Interpreter右側的小齒輪 --> 選擇Add Python Interpreter：
在彈出的界面中選擇第二個Conda Environment，選中Existing Environment,設置路徑為anaconda安裝路徑下的python.exe,如下圖：
如此，即完成了對gpu版本pytorch的配置，要驗證是否安裝成功，在pycharm的Terminal輸入代碼：
1 python 2 import torch 3 torch.cuda.is_available()
若輸出結果為True，則pytorch-gpu安裝成功。