Window下Python+CUDA+PyTorch安裝
1 概述
Windows下Python+CUDA+PyTorch安裝，步驟都很詳細，特此記錄下來，幫助讀者少走彎路。
2 Python
Python的安裝還是比較簡單的，從官網下載exe安裝包即可：
因為目前最新的torch版本只支援到Python 3.8，因此為了不會出現版本相容問題建議安裝Python 3.8。
下載中間的那個executable installer並安裝，開啟會建議選擇自定義：
選擇需要的元件：
選擇位置：
安裝完後可以從命令列檢視是否安裝成功（一般預設會把路徑新增到環境變數中）：
python --version
由於該安裝包會自動把pip也一起安裝，因此在使用時會提示pip
pip：
python -m pip install --upgrade pip
在安裝前建議新增國內源，在使用者目錄下新建一個pip資料夾，並在其中新建一個叫pip.ini的檔案，輸入如下內容：
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host = https://pypi.tuna.tsinghua.edu.cn
3 CUDA
CUDA的安裝也不難，一般Windows會在裝系統的時候自動安裝好對應的N卡驅動，從官網下載對應的CUDA Toolkit
下載好後開啟（實際上是一個壓縮包，這裡是解壓路徑）：
解壓完成後會自動開啟安裝介面：
沒特殊需求可以直接使用預設安裝選項，否則請選擇自定義元件進行安裝：
由於筆者沒有安裝VS，這裡會提示找不到VS，忽略即可：
接下來就是等待安裝：
安裝完成：
開啟cmd輸入nvcc --version，顯示版本號即為安裝成功：
4 PyTorch
4.1 下載
PyTorch建議使用離線方式手動安裝，戳這裡下載對應CUDA版本與對應Python版本的torch以及torchvision：
最前面的cu110表示cuda 11.0，接著是torch/torchvision，再接著是版本，後面的cp
Python版本，比如cp36表示Python 3.6，接著是系統和架構。
4.2 安裝
下載完成後使用pip安裝即可：
pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl
注意如果是Windows 2004版本可能會出現如下問題：
這是numpy 19.0.4版本的問題，詳情可以戳這裡。
解決方法是安裝19.0.3版本：
pip uninstall numpy
pip install numpy==19.0.3
接著進行測試，開啟python：
python
import torch
print(torch.__version__)
輸出torch版本即表示成功：