win10下pytorch-gpu安装以及CUDA详细安装过程

win10下pytorch-gpu安装以及CUDA详细安装过程win10下pytorch-gpu安装以及CUDA详细安装过程1.Cuda的下载安装及配置首先我们要确定本机是否有独立显卡。在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡。可以看到本机有一个集成显卡和独立显卡NVIDIAGetForceGTX1050。接下来,测试本机独立显…

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺

           win10下pytorch-gpu安装以及CUDA详细安装过程

1.Cuda的下载安装及配置

        首先我们要确定本机是否有独立显卡。在计算机-管理-设备管理器-显示适配器中,查看是否有独立显卡。

win10下pytorch-gpu安装以及CUDA详细安装过程

            可以看到本机有一个集成显卡和独立显卡NVIDIA GetForce GTX 1050。

            接下来,测试本机独立显卡是否支持CUDA的安装,点击此处查询显卡是否在列表中。

win10下pytorch-gpu安装以及CUDA详细安装过程

win10下pytorch-gpu安装以及CUDA详细安装过程

          从上图中,可以看到我本机的独立显卡是支持CUDA安装的,计算力是6.1。

          那么,接下来就是选择CUDA进行安装即可,点击此处

win10下pytorch-gpu安装以及CUDA详细安装过程

 

win10下pytorch-gpu安装以及CUDA详细安装过程

        这里我选了CUDA Toolkit10.0的版本,至于选择哪个版本,个人认为应该没多大差别,一般就是看这个版本是否要求GPU的计算能力是多少以上。大约是2.1G。我用windows的IDM下载工具下载的,速度蛮快的。

       接着就是安装过程,双击打开显示临时解压目录,不需要改变,默认即可。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

         接下来,进入NVIDIA安装过程,在这安装过程中,我一开始直接选择的精简安装,但由于VS的原因,导致无法正常安装,于是我换成了自定义的安装方式,并将VS勾给去掉,便可以正常安装了,至于CUDA的安装目录,大家默认安装在C盘即可。

win10下pytorch-gpu安装以及CUDA详细安装过程

win10下pytorch-gpu安装以及CUDA详细安装过程

 

 

        安装完成之后,便是配置环境变量。环境变量配置如下图所示。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

       测试CUDA是否正常安装。可以看到测试成功,接着进行cudnn的下载。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

2.CUDNN的下载及配置

        点击此处,选择下载download cudnn,但这里需要你注册一个账号,然后进行问卷之后才可以进行下载页面,反正一步步操作即可。

       然后因为我上一步CUDA的版本是10.0,而CUDNN的版本要跟CUDA版本一致,所以选择第二个下载即可。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

         下载之后,解压缩,将CUDNN压缩包里面的bin、clude、lib文件直接复制到CUDA的安装目录下,直接覆盖安装即可。

win10下pytorch-gpu安装以及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

win10下pytorch-gpu安装以及CUDA详细安装过程

 

       然后命令行验证pytorch是否正常安装成功,这里可以正常打印出版本号,安装没问题。

import torch

print(torch.__version__)

 

win10下pytorch-gpu安装以及CUDA详细安装过程

 

         但接下来我尝试让其输出是否可以正常调用CUDA,却输出False.

print(torch.cuda.is_available())

        经过一段时间的交流,找到问题所在。因为我本机是集成显卡和独立显卡并存,可能是因为默认调用集成显卡而造成的问题,那么接下来就是将独立显卡设为默认即可。

        但我因为多次尝试的结果,成功将我本机的NAVIDIA控制面板给卸载掉了,无法正常打开,所以我还需要重新下载NVIDIA的控制面板,来到官网此处,选择对应的版本下载安装即可。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

         然后右键打开NIVIDIA控制面板,选择管理3D设置,全局设置中选择高性能NVIDIA处理器即可。

win10下pytorch-gpu安装以及CUDA详细安装过程

win10下pytorch-gpu安装以及CUDA详细安装过程

 

 

        然后在命令行下重新输入

import torch

print(torch.cuda.is_available())

       即可返回True,若还返回False的话 ,记得重启一下电脑,遇到问题重启一下电脑,看是否可以正常显示。

win10下pytorch-gpu安装以及CUDA详细安装过程

 

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/190417.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)


相关推荐

  • Java课程设计【学生信息管理系统】[通俗易懂]

    Java课程设计【学生信息管理系统】[通俗易懂]课程设计目录一、问题描述二、基本要求三、需求分析四、概要设计1、类之间的调用关系2、学生信息模块3、管理系统模块4、详细设计①主程序LoginGUI的代码②程序View的代码③程序Student的代码④程序ConnectSQLServer的代码五、调试分析六、用户使用说明1、登录2、添加3、查询4、修改5、删除6、退出七、测试结果八、课程设计总结九、参考文献一、问题描述如何实现一个功能简单的学生信息管理系统,能够对学生信息(包括照片)进行添加、删除、修改和查询等操作。二、基本要求实现一个功能简单的学

    2022年10月17日
  • python hashlib模块安装_python hashlib 模块

    python hashlib模块安装_python hashlib 模块hashlib模块用于加密相关的操作,代替了md5模块和sha模块,主要提供SHA1,SHA224,SHA256,SHA384,SHA512,MD5算法。在python3中已经废弃了md5和sha模块什么是摘要算法呢?摘要算法又称哈希算法、散列算法。它通过一个函数,把任意长度的数据转换为一个长度固定的数据串(通常用16进制的字符串表示)。摘要算法就是通过摘要函数f()对任意长度的数据data计算出…

  • 中国名花异草故事

    中国名花异草故事中国名花异草故事作者:刘桂莉、余绮中国名花异草故事—总序  我不是旅行家,几十年来到过的地方却不少。每到一地,工作之余,对当地的一些人情、习俗、山川、风物,以至特殊的传说故事,都颇感兴趣。在谈论或介绍本地的特产时,当地人往往绘声绘色而且充满着喜悦与自豪,连带讲到的一些掌故与传奇,我也感觉兴味盎然。因为其中有些内容,活泼生动,不仅使我扩大了见闻,觉得新鲜有趣,还能增进我对这些地方历史、社会的

  • ioctl函数详解_lseek函数

    ioctl函数详解_lseek函数 我这里说的ioctl函数是在驱动程序里的,因为我不知道还有没有别的场合用到了ioctl,所以就规定了我们讨论的范围。为什么要写篇文章呢,是因为我前一阵子被ioctl给搞混了,这几天才弄明白它,于是在这里清理一下头脑。一、什么是ioctl。ioctl是设备驱动程序中对设备的I/O通道进行管理的函数。所谓对I/O通道进行管理,就

    2022年10月17日
  • pymssql for linux[通俗易懂]

    pymssql for linux[通俗易懂]linux连接SQLServer,pymssql安装包下载,链接:https://pan.baidu.com/s/1zXyhvatpoaFRpcptmv0reA密码:y3w1安装如下:yuminstallpython-devel-ytarzxfpytz-2018.4.tar.gzcdpytz-2018.4pythonsetup.pyinstallcdta…

  • springMVC统一日志处理和异常拦截

    springMVC统一日志处理和异常拦截springMVC统一日志处理和异常拦截

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号