前言
現在使用 pytorch 框架進行深度學習的人越來越多了，但是 torch 的安裝（GPU 版本）和普通第三方包的安裝過程不太一樣，因為涉及到電腦的顯卡驅動版本，下面來看看安裝 pytorch 的正確姿勢。
安裝過程
查看驅動版本
打開 NVIDIA 控制面板，點擊工具欄幫助 --> 系統信息，在彈出的對話框中有如下信息：
可以看到本機驅動程序版本為 466.92，接着前往 NVIDIA CUDA Toolkit Release Notes，在 cuda driver 這一節有下面這個表：
由於本機為 Windows 系統且驅動版本為 466.92，所以應該選擇 CUDA 11.6.x。
下載 torch
在 PyTorch 官網的 Get Started 頁面有下圖所示的安裝向導：
雖然沒找到 CUDA 11.6.x，但是 CUDA 11.3 也是可以的，復制 Run this Command 里面的指令：
pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
這里把 pip3 換成了 pip，不過 pip3 也是可以的。在終端運行上述指令，等待安裝完成即可。