参考：
https://www.jianshu.com/p/59b6b23c3d11
1.查看当前NVIDIA显卡对应的驱动版本：
2.安装CUDA9.0（旧版）
https://developer.nvidia.com/cuda-toolkit-archive
新版本直接上官网下载：
https://developer.nvidia.com/cuda-downloads
3.下载cuDNN（对应CUDA版本）
https://developer.nvidia.com/rdp/cudnn-archive
安装方法就是将对应文件夹内文件拷到cuda安装路径下。
安装：
https://pytorch.org/get-started/locally/
5.最后测试：
完成。