首页 > 其他 > 详细

深度学习环境配置: 英伟达RTX2060 + CUDA 9.2 + cuDNN 7.2.1

时间:2019-05-16 15:05:54      阅读:1690      评论:0      收藏:0      [点我收藏+]

前言

拿到了新本本,就迫不及待想要开始配置好torch的深度学习环境了. 本本是MECHREVO深海系列的Z2-R, CPU i7-9750, 独立显卡英伟达RTX-2060 (性价比还算不错, 再加上本身的轻薄游戏本特性, 对我这种天天码代码偶尔打dota2的太合适了). 这篇博客主要记录CUDA 10.0 + cuDNN 7.5.0的安装配置流程.

资源下载

  1. CUDA 10.0 : 官网下载地址       (注意, 如果不挂梯子, 安装包文件会一直下载失败, 注意,梯子要用全局模式;网络安装版,可以下载,但是无法安装。)
  2. cuDNN 7.5.0 : 官网下载地址    (这里面有好几个for CUDA9.2的,一定要注意,因为关系到后面torch的版本,这里出错了,后面就会报错!!!)
  3. VS 2015 社区版: 下载地址   (注意, 这个链接点击便直接开始下载了)
  4. pytorch 1.0 GPU版: 官网下载地址 

CUDA安装

在安装CUDA的时候提示安装失败,在检索反馈信息之后,我发现 Visual Studio Integration 这个软件安装失败,这个错误导致安装程序无法正常的继续下去。

解决方案步骤如下:

- 重新安装CUDA,注意选择自定义安装

- 在选择安装类别的时候,取消选择Visual Studio Integration; 并且只安装CUDA组件, 取消其他的NVIDIA驱动安装 (可能比你自己电脑的驱动要旧)

- 安装完成

- 将CUDA安装包解压,将CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions 下的所有文件,复制到VS的对应目录下,参考路径如下

C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations

到此为止CUDA就安装完成了,

( CUDA 9.2: 注意在最后的时候需要运行一下补丁安装包,这个补丁修复了CUDA9.2某个算法库的缺陷。)


深度学习环境配置: 英伟达RTX2060 + CUDA 9.2 + cuDNN 7.2.1

原文:https://www.cnblogs.com/muhanxiaoquan/p/10875374.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!