win10+NVIDIA geforce gtx 960M+cuda9.0+cudnn7.1+pytorch_1.1.0+torchvision_0.3.0---已解决

今天折腾了一天，终于是把它折腾出来了，可能真的是自己太菜了，废话不多说，简单记录一下。

配置

1. 系统 ：win10
2. python版本 ：3.7.3
3. 显卡 ：NVIDIA geforce gtx 960M
4. 安装cuda版本 ：cuda9.0
5. cudnn版本 ：cudnn7.1
6. pytorch版本 ：pytorch_1.1.0
7. 编译工具：pycharm

安装步骤

1.安装cuda

（1） 首先检擦自己电脑显卡是否支持cuda，这一点网上有很多，没啥营养，大致就是打开设备管理器，点击显示适配器，然后找到自己的显卡，查看是否安装NVIDIA显卡。若安装，在桌面右键打开NVIDIA控制面板，点击右下角的系统信息，点击组件，在列表中查看自己显卡可支持的cuda版本，如图：

也可以对照这个表（NVIDIA也给出了对应关系：传送门）查找可以安装的cuda toolkit的版本（比如我的程序版本号是385.54，因此选择cuda9.0）。
注意，这个驱动是向下兼容的，因此这也决定了其可安装的最高版本。

（2） 下载cuda
从NVIDIA的官网下载对应版本的cuda toolkit，传送门。
并进行以下选择进行下载：

（3） 安装cuda，双击.exe安装包。

会进入安装界面：

选择同意并继续。
之后弹出安装选项，选择自定义模式