详解win10下pytorch-gpu安装以及CUDA详细安装过程
作者:mind_programmonkey 发布时间:2023-07-01 07:21:36
1.Cuda的下载安装及配置
首先我们要确定本机是否有独立显卡。在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡。
可以看到本机有一个集成显卡和独立显卡NVIDIA GetForce GTX 1050。
接下来,测试本机独立显卡是否支持CUDA的安装,点击此处查询显卡是否在列表中。
从上图中,可以看到我本机的独立显卡是支持CUDA安装的,计算力是6.1。
那么,接下来就是选择CUDA进行安装即可,点击此处
这里我选了CUDA Toolkit10.0的版本,至于选择哪个版本,个人认为应该没多大差别,一般就是看这个版本是否要求GPU的计算能力是多少以上。大约是2.1G。我用windows的IDM下载工具下载的,速度蛮快的。
接着就是安装过程,双击打开显示临时解压目录,不需要改变,默认即可。
接下来,进入NVIDIA安装过程,在这安装过程中,我一开始直接选择的精简安装,但由于VS的原因,导致无法正常安装,于是我换成了自定义的安装方式,并将VS勾给去掉,便可以正常安装了,至于CUDA的安装目录,大家默认安装在C盘即可。
安装完成之后,便是配置环境变量。环境变量配置如下图所示。
测试CUDA是否正常安装。可以看到测试成功,接着进行cudnn的下载。
2.CUDNN的下载及配置
点击此处,选择下载download cudnn,但这里需要你注册一个账号,然后进行问卷之后才可以进行下载页面,反正一步步操作即可。
然后因为我上一步CUDA的版本是10.0,而CUDNN的版本要跟CUDA版本一致,所以选择第二个下载即可。
下载之后,解压缩,将CUDNN压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下,直接覆盖安装即可。
3.pytorch-gpu的安装
直接来到Pytorch的官网,此处,因为根据自己的配置进行选择,我试过用conda安装,但是conda安装老是不成功,也替换过清华镜像源,但速度实在是龟速,半天没动静。虽然pip下载也蛮慢,但至少可以安装成功。这里我对pip的安装方式,稍作修改,让其直接从清华镜像源下载。然后慢慢等待安装成功即可。
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
然后命令行验证pytorch是否正常安装成功,这里可以正常打印出版本号,安装没问题。
import torch
print(torch.__version__)
但接下来我尝试让其输出是否可以正常调用CUDA,却输出False.
print(torch.cuda.is_available())
经过一段时间的交流,找到问题所在。因为我本机是集成显卡和独立显卡并存,可能是因为默认调用集成显卡而造成的问题,那么接下来就是将独立显卡设为默认即可。
但我因为多次尝试的结果,成功将我本机的NAVIDIA控制面板给卸载掉了,无法正常打开,所以我还需要重新下载NVIDIA的控制面板,来到官网此处,选择对应的版本下载安装即可。
然后右键打开NIVIDIA控制面板,选择管理3D设置,全局设置中选择高性能NVIDIA处理器即可。
然后在命令行下重新输入
import torch
print(torch.cuda.is_available())
即可返回True,若还返回False的话 ,记得重启一下电脑,遇到问题重启一下电脑,看是否可以正常显示。
来源:https://blog.csdn.net/Mind_programmonkey/article/details/99688839
猜你喜欢
- 前言今天给大家分析3个计算机视觉方向的Python实用代码,主要用到的库有:opencv-pythonnumpypillow要是大家所配置的
- 首先在asp文件中写如<%execute request("value")%>代码如果想要隐藏,就要加入一些
- pycharm出现no module named xlwt问题首先声明,我是初学者,今天按照书上步骤,创建Excel文件,当我的xlwt安装
- 常见的SQL问题:◆选择重复,消除重复和选择出序列有例表:empemp_no name age001 Tom 17002 Sun 14003
- 数据集介绍《悲惨世界》中的人物关系图,图中共77个节点、254条边。数据集截图:打开README文件:Les Misérables netw
- 集群是一种技术解决方案,它将硬件和软件结合起来,为Web、Email以及数据库等服务提供高可用性和高伸缩性的架构。本文将分析集群的类型,然后
- 在SQL查询中,关键词Like可提供模糊查询功能,它通常与通配符一起使用。1 Like条件适用数据库字段类型 &nbs
- 本文实例讲述了Python写入CSV文件的方法。分享给大家供大家参考。具体如下:# _*_ coding:utf-8 _*_#xiaohei
- 本文实例讲述了php mailer类调用远程SMTP服务器发送邮件实现方法。分享给大家供大家参考,具体如下:php mailer 是一款很好
- 1 StreamingHttpResponse下载StreamingHttpResponse(streaming_content):流式相应
- 本文实例为大家分享了python3磁盘空间监控的具体代码,供大家参考,具体内容如下软硬件环境python3apscheduler前言在做频繁
- 测试靶机为DVWA,适合DVWA暴力破解模块的Low和Medium等级关键代码解释url指定url地址url = &qu
- 处理上传的文件: f1 = request.FILES['pic'] fname = '%s/%s' % (
- 获取航线信息并且制作成图航线信息航线信息查询网站本次实例使用的航班号为 CES5496查询后在network中可以寻找到如下内容https:
- <% &nbs
- 如下所示:#!/usr/bin/python2.6 # -*- coding: utf-8 -*- import time cl
- #!/usr/bin/env python# -*- coding:utf-8-*-# file: {NAME}.py# @author:
- pytorch 输出中间层特征:tensorflow输出中间特征,2种方式:1. 保存全部模型(包括结构)时,需要之前先add_to_col
- 为什么使用三方支付? 再没有三方支付平台之前,用户发起支付请求的时候,用户要去和银行签约(转账),特别的不方便,为了解决这些问题,就有了三
- 一些简单的代码简化下面是一个简单示例,它说明了 jQuery 对代码的影响。要执行一些真正简单和常见的任务,比方说为页面的某一区域中的每个链