安装cuda + pytorch 的注意事项

1. 需要确定电脑上的显卡是否支持cuda，当前的显卡是支持哪个cuda版本。在linux版本上，需要给显卡安装nvidia驱动，然后运行nvidia-smi指令查看显卡驱动信息。其结果可能如下所示

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.50       Driver Version: 430.50       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GT 730      Off  | 00000000:01:00.0 N/A |                  N/A |
| 52%   43C    P0    N/A /  N/A |    695MiB /   974MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0                    Not Supported                                       |
+-----------------------------------------------------------------------------+

这样的信息就表示当前显卡驱动支持的是cuda 10.1版本

2. 在安装pytorch的时候，需要安装与显卡当前同样cuda版本的软件包。在pytorch上提供的可以安装cuda版本比较多，一般首页提示上只有最新的几个版本。如果这个没有电脑上显卡nvidia-smi指令看到的版本，则需要去pytorch的历史版本上去找。https://pytorch.org/get-started/previous-versions/