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

最新推荐文章于 2024-07-09 14:43:01 发布

置顶

喵喵love

最新推荐文章于 2024-07-09 14:43:01 发布

阅读量6.1k

点赞数 15

分类专栏： Pytorch 文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/qq_41603193/article/details/111599769

版权

本文详细记录了在Win10系统上，使用NVIDIA geforce gtx 960M显卡，安装CUDA9.0、cudnn7.1以及PyTorch1.1.0的过程，包括解决安装过程中遇到的问题，如CUDA的Visual studio integration安装失败，以及离线安装PyTorch和Torchvision的方法。最后介绍了如何在PyCharm中配置conda虚拟环境pytorch-gpu。

摘要由CSDN通过智能技术生成

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

配置

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

安装步骤

1.安装cuda

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

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

（3）安装cuda，双击.exe安装包。
在这里插入图片描述
会进入安装界面：

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