CUDA和Pytorch安装记录

1、关键点：

显卡算力 < cuda runtime version <= cuda driver version

1. 显卡算力：根据显卡型号查表获得。https://en.wikipedia.org/wiki/CUDA
2.  cuda runtime version：GPU的编程平台版本。
3.  cuda driver version：显卡驱动版本。为了保险起见，应该更新到最新版，留足裕量。https://www.nvidia.cn/Download/index.aspx?lang=cn

2. Pytorch安装：

2.1：CUDA安装

如果是用GPU版的pytorch，首先应该安装cuda 。

1、打开prompt，输入指令：nvidia-smi

右侧12.5为最高支持版本号。

2、输入指令：nvidia-smi

安装之后会显示版本号，若未安装，则显示未找到命令。通常只安装显卡驱动的话，cuda是没安装的。

  cuda-downloads

2.2：CUDNN安装

安装完了CUDA之后还需要安装CUDNN，

cudnn-downloads

 

---

以下还未实操， 仅作记录。

---

2.3：pytorch和torchvision安装

这里需要注意版本匹配，

cuda，cudnn，pytorch，torchvision的版本对应关系

官方网站的匹配工具：https://pytorch.org/get-started/locally/

测试pytorch是否安装成功
我们打开cmd输入如下命令

python
import torch
torch.__version__

待定