1 概述
Windows
下
Python+CUDA+PyTorch
安裝，步驟都很詳細，特此記錄下來，幫助讀者少走彎路。
2 Python
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
CUDA
的安裝也不難，一般
Windows
會在裝系統的時候自動安裝好對應的
N卡
驅動，從官網下載下傳對應的
CUDA Toolkit
下載下傳好後打開（實際上是一個壓縮包，這裡是解壓路徑）：
解壓完成後會自動打開安裝界面：
沒特殊需求可以直接使用預設安裝選項，否則請選擇自定義元件進行安裝：
由于筆者沒有安裝
VS
，這裡會提示找不到
VS
，忽略即可：
接下來就是等待安裝：
安裝完成：
打開
cmd
輸入
nvcc --version
，顯示版本号即為安裝成功：
4 PyTorch
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
版本即表示成功：