windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

2024-07-16 1828阅读

超详细介绍安装Gpu版本的pytorch深度学习环境

  • 一、显卡驱动安装
    • 1.1 下载驱动
    • 1.2 安装驱动
    • 二、cuda安装
      • 2.1 下载cuda
      • 2.2 安装cuda
      • 2.3 检查cuda是否安装成功
      • 三、安装cudnn
        • 3.1 cudnn下载
        • 3.2 cudnn安装
        • 四、安装miniconda
          • 4.1 miniconda下载
          • 4.2 miniconda安装
          • 4.3 添加环境变量
          • 五、安装pytorch和torchvision
            • 5.1 下载pytorch和torchvision
            • 5.2 pytorch和torchvision安装
            • 5.3 检查GPU版本的torch是否可用
            • 六、安装pycharm
              • 6.1 pycharm下载
              • 6.2 pycharm安装
              • 6.3 pycharm中配置conda环境
              • 七、附本人所使用的安装包截屏

                一、显卡驱动安装

                大多数windows10/Windows11其实已安装好了显卡驱动,如果你已经安装好了显卡驱动,请直接跳过该章。

                1.1 下载驱动

                进入官方下载地址 https://www.nvidia.cn/drivers/lookup/

                根据自己显卡的类型,选择对应的参数,最后单击查找并下载即可。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                1.2 安装驱动

                双击下载下来的驱动,直接安装即可,尽量选择默认安装路径。安装完成后,可以按win+r键盘,输入cmd回车,弹出如下所示终端,输入nvidia-smi,若显示如下,则说明驱动已经安装好了。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                二、cuda安装

                安装cuda时,注意版本对应问题,下图中的1表示你可以安装的cuda的最高版本。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                2.1 下载cuda

                https://developer.nvidia.com/cuda-toolkit-archive

                我可以装的最高版本是12.5,我下载12.1.0版本的cuda(主要是为了能下载到对应版本的pytorch,版本太高,对应pytorch都很难下载到哦)

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                按照图中所示,选择自己的相关信息,然后下载,我的信息选择如下图所示:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                2.2 安装cuda

                双击所下载的cuda安装包,然后选择安装的目录,本人默认,然后单击OK,接着一路的选择如以下图所示,按照顺序依次走完就安装完成了。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                2.3 检查cuda是否安装成功

                打开终端,输入nvcc -v,弹出如下界面,本人安装的cuda版本为12.1.0,安装成功。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                三、安装cudnn

                3.1 cudnn下载

                https://developer.nvidia.com/rdp/cudnn-archive

                本人下载的如下图所示,只要和cuda版本对应即可。注意,下载时需要注册并登录账号才可以。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                3.2 cudnn安装

                解压掉下载下来的cudnn后,如下图所示:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                将上图所示中的三个文件,依次粘贴到cuda的对应目录中,我的cuda安装目录为:

                C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1
                

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                说详细一点就是:

                1、把解压后cudnn文件中bin下的所有内容粘贴到:

                C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin
                

                2、把解压后cudnn文件中include下的所有内容粘贴到:

                C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\include
                

                3、把解压后cudnn文件中lib\x64下的所有内容粘贴到:

                C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\lib\x64
                

                四、安装miniconda

                4.1 miniconda下载

                https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/?C=S

                我下载的是如下图所示的版本:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                4.2 miniconda安装

                双击Miniconda安装包,后面所有安装过程依次见以下图系:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                4.3 添加环境变量

                将以下minionda的三个文件路径,添加到环境变量当中去,我的路径如下:

                C:\ProgramData\miniconda3\Library\bin
                C:\ProgramData\miniconda3
                C:\ProgramData\miniconda3\Scripts
                

                添加环境变量的过程为:首先光标移到我的电脑上右键–》属性》高级系统设置后,弹出窗口如下,按照图中标号选择操作,接着依次按后面图序操作即可。

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                双击path

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                至此miniconda安装完成。

                五、安装pytorch和torchvision

                5.1 下载pytorch和torchvision

                pytorch:https://download.pytorch.org/whl/torch/

                torchvision:https://download.pytorch.org/whl/torchvision/

                本人下载的torch和torchvision的版本如下图所示:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                5.2 pytorch和torchvision安装

                打开终端,创建使用conda创建环境并激活

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                设置pip镜像地址,不然安装torch时自动下载其他依赖的过程很慢:

                pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
                

                进入到torch及torchvision所在的文件夹,使用pip安装

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                5.3 检查GPU版本的torch是否可用

                使用如下命令查看GPU版本的pytorch是否可用

                import torch
                torch.cuda.is_available()
                

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                从图中可以看到,完美可用。

                六、安装pycharm

                6.1 pycharm下载

                https://www.jetbrains.com.cn/pycharm/download/?section=linux

                进入后如下如,如果要下载社区免费版本就向下滑动,社区版本就在本页最下面,选好后会直接下载的

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                6.2 pycharm安装

                双击安装包,依次按如下图所示操作,即可完成安装:

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                6.3 pycharm中配置conda环境

                用pycharm打开或者新建一个项目,接着按照以下图序继续往下走

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

                七、附本人所使用的安装包截屏

                windows中超详细深度学习环境配置之安装显卡驱动、cuda、cudnn、pytorch、torchvision、pycharm

VPS购买请点击我

免责声明:我们致力于保护作者版权,注重分享,被刊用文章因无法核实真实出处,未能及时与作者取得联系,或有版权异议的,请联系管理员,我们会立即处理! 部分文章是来自自研大数据AI进行生成,内容摘自(百度百科,百度知道,头条百科,中国民法典,刑法,牛津词典,新华词典,汉语词典,国家院校,科普平台)等数据,内容仅供学习参考,不准确地方联系删除处理! 图片声明:本站部分配图来自人工智能系统AI生成,觅知网授权图片,PxHere摄影无版权图库和百度,360,搜狗等多加搜索引擎自动关键词搜索配图,如有侵权的图片,请第一时间联系我们,邮箱:ciyunidc@ciyunshuju.com。本站只作为美观性配图使用,无任何非法侵犯第三方意图,一切解释权归图片著作权方,本站不承担任何责任。如有恶意碰瓷者,必当奉陪到底严惩不贷!

目录[+]