首先要明确gpu是哪个显卡并且已安装的显卡驱动或者将要安装哪个版本,nVidia官网https://www.nvidia.com/Download/index.aspx,

1. 安装CUDA10.1

先查看本机显卡适合哪个版本的CUDA(邻近版本有时也可以,如果项目需求和电脑要求不一样,可以尝试一下可不可以),打开“控制面板”---》“硬件与声音”---》“NVIDIA控制面板”---》“帮助”----》“系统信息”--->“组件”,如图二红线所示,我系统适合CUDA10.1。

注:

最主要还是去nVidia官网 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 查看cuda和显卡驱动对应表

知道适合哪个版本的CUDA,就可以去官网下载对应的CUDA了,但是官网首页的CUDA一般是最新版,我们可能需要下载旧版本,https://developer.nvidia.com/cuda-toolkit-archiv,进入这个网址下载可以看到旧版本列表,如下图所示。

点击相应版本,进入下图界面,选择OS,系统64位,版本,然后选择exe(local)是一个完整的包,exe(network)得自己安装组件,下载完CUDA,然后进行安装就行。点击相应版本,进入下图界面,选择OS,系统64位,版本,然后选择exe(local)是一个完整的包,exe(network)得自己安装组件,下载完CUDA,然后进行安装就行,安装完成,在win+R输入cmd打开命令行输入 nvcc -V,如下图所示安装成功。

注:如果CUDA是自定义安装,一般系统不会自动给CUDA配置环境变量,需要手动配置,此时,nvcc -V命令没有正常输出。通过在系统变量path中添加CUDA目录下的bin和libnvvp两个目录即可,如下图。

2. 安装cudnn

下载cudnn,https://developer.nvidia.com/rdp/cudnn-archive 进入各个旧版本的cudnn的网址,选择相对应CUDA的cudnn版本,一个CUDA有多个cudnn版本对应,随便选一个即可

② 下载后,将压缩包解压得到cuda文件夹,文件夹下有三个文件夹,复制这三个文件夹到目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1下,会自动将cudnn的三个文件夹的文件合并到CUDA\v10.1的三个同名文件夹bin、include和lib中,如下图所示。

3. 安装Pytorch

https://pytorch.org/get-started/locally/,进入pytorch官网进行系统,语言,安装方式的选择进行安装。要想得到安装其它旧版本的命令,可以点击下图中蓝色线处转到旧版本页面。

推荐使用pip安装,如果有安装annaconda,选择conda就行,然后选择相应CUDA版本,得到了下图红线所示的命令,在Windows命令行输入命令即可。

pytorch旧版本页面

注意:虽然图中pytorch1.5.1可以匹配CUDA10.1,但是我安装pytorch1.5.1和pytorch1.4.0都没有安装成功,会有下图所示的错误。

最后安装pytorch1.3.0成功。但是cuda9.2版本的pytorch,我前面安装的cuda是10.1,所以如果官网的安装命令和版本有很多莫名其妙问题的时候,可以自己多尝试其它的安装版本。如直接在pip install时指定torch==1.3.0来指定版本,然后torchvision的话可以安装torch时自动安装,若没有自动安装,则手动pip install安装。

注:

手动安装可以直接转去pytorch的whl网页https://download.pytorch.org/whl/cu101/torch_stable.html,比如cuda10.1版本就去cu101,cuda9.2版本就去cu92网页,然后下载whl文件后,直接pip install torch-1.1.0-whl(改成你的文件名)就行。

简单测试是否安装成功,可通过在命令行或终端import torch这个包

然后最终测试是否成功可通过简单运行以下代码测试

Windows 10安装CUDA10.1+cudnn7.6.0+Pytorch1.3.0相关推荐

  1. Windows 10安装TensorFlow-gpu1.4 及CUDA8.0,cuDNN6.0,搞定了,包含安装方法和下载路径

    Windows 10安装TensorFlow-gpu1.4 及CUDA8.0,cuDNN6.0,搞定了 先说下我的最终版本: Python 3.5.3: tensorflow-gpu==1.4.0 : ...

  2. Windows 10 安装FileZilla server 中文版 0.9.46 局域网FTP

    Windows 10 安装FileZilla server 中文版 0.9.46 局域网FTP FileZilla server 中文版 下载地址:https://filezilla-project. ...

  3. Windows 10 安装 NVIDIA cuDNN v7.6.0 for CUDA 10.1

    Windows 10 安装 NVIDIA cuDNN v7.6.0 for CUDA 10.1 NVIDIA 引领人工智能计算 - NVIDIA https://www.nvidia.cn/ NVID ...

  4. Windows 10 安装使用TensorFlow-GPU

    Windows 10 安装使用TensorFlow-GPU 当前环境描述:Win10 64位,Python3.6 目标:安装使用TensorFlow1.6-GPU 所需文件:Cuda9.0.Cudnn ...

  5. windows 10 安装和使用中5个常见问题

    2015年7月29日0点起,Windows 10推送全面开启,Windows7.Windows8.1用户可以免费升级到Windows 10,用户也可以通过系统升级到Windows10,在这过程中,用户 ...

  6. Windows 10 安装 scrapy 步骤

    1. 安装 twisted 打开 cmd 输入 python 命令查看自己的 Python 版本,如图我的版本是 Python 3.8: 打开 https://www.lfd.uci.edu/~goh ...

  7. Ubuntu16.04装机2:安装CUDA10.2+cuDNN7.6.5

    本篇文章是在你的台式电脑已经安装好NVIDIA驱动的前提下继续安装CUDA10.2的教程.nvidia安装教程见: 文章转自:Ubuntu16.04安装CUDA10.2+cuDNN7.6.5(福利区) ...

  8. 从头开始 windows 10 安装awscli,aws sam

    安装软件尽量用pip安装 1.windows 10安装了python 3.8.0 升级pip到最新版本 2.windows 10 安装django 3.安装了awscli pip install aw ...

  9. Windows 10 安装STF实战

    Windows 10 安装STF实战 智慧rdda大成 AI开发测试实战 1周前 STF 手机设备管理平台 docker 安装stf docker pull openstf/stf:latest  / ...

最新文章

  1. SAP QM 执行事务代码QS23为检验特性分配Selected Set的时候报错 - You cannot use entries from catalogs 1 and 3-
  2. SAP最佳业务实践:重复制造(149)-4发料
  3. windows 建立wifi热点
  4. 解决:No goals have been specified for this build. You must specify a valid lifecycle phase or a goal i
  5. DataGridView 设置行不可见时,与货币管理器的位置关联的行不能设置为不可见
  6. 编译的警告:隐式声明与内建函数 ‘malloc’ 不兼容
  7. linux环境判断nginx是否已经启动
  8. php-mvc模式(2)
  9. tar zip 打包相关操作
  10. python语音识别_Python语音识别终极指南
  11. python中返回上上级目录的命令
  12. 手机安全卫士(四)----- Splash界面
  13. 为什么电子元器件会老化?像人变老一样吗?
  14. 【译】Linux不同的IO访问方式中,Scylla的选择和依据
  15. 安卓开发自己写的刻度尺测量,精确到mm.
  16. YOLOv3中的非极大值抑制
  17. 运动模糊 motion blur
  18. 请主动远离打击你自信的人
  19. java 图片合并成pdf_java将多张图片合并转为PDF
  20. 各种CAD工程常用图例,如何轻松看懂图!

热门文章

  1. sap scc4 客户端设置
  2. 冬天到了:你检查过你的防冻液了吗!
  3. SAP系统与外部系统的三种接口形式
  4. 如何在ALV中输出标题头
  5. 消息转为文本函数('MESSAGE_TEXT_BUILD')
  6. 根目录_Linux Shell从入门到删除根目录跑路指南
  7. java占位符打印_java简单日志打印规范小记
  8. python接口测试框架设计_Python3简易接口自动化测试框架设计与实现(中)
  9. float 精度_float相加产生精度损失的原因是什么?
  10. vue操作dom_vue源码全面解析(四十六)源码中操作DOM的方法集合