cuda+cuDNN+pytorch_gpu安装

本文详细指导了如何在Windows上安装PyTorchGPU版本,包括更新Anaconda,选择并安装CUDA、cuDNN,设置环境变量以及创建和激活conda虚拟环境,特别提到CUDA版本选择与cuDNN的对应关系。
摘要由CSDN通过智能技术生成

一、前言

前期已安装好anaconda和pycharm,安装pytorch_gpu版本涉及的一些问题如下:

  1. anaconda最好保持最新更新
  2. cuda是nvidia提供的显卡驱动模块,cuda的安装需要对应电脑显卡cuda的版本进行选择,可向下兼容
  3. cuDNN是nvidia提供的cuda的深度学习加速插件
  4. pytorch_GPU版本的选择需对应cuda的版本,列表中查询不到对应版本向下兼容
  5. 换源下载,使用国内清华镜像源

二、安装cuda

  • 查看cuda版本

进入cmd,输入以下指令

nvcc -V     # 当前正在使用的cuda版本
nvidia-smi  # 获取显卡最高支持cuda版本

 nvvv -V查看显卡目前应对应的cuda版本

nvidia-smi是获取显卡最高支持cuda版本,并查看当前显卡驱动版本信息

选择对应的11.5cuda版本,选择windows系统,64位,系统版本win11,选择本地安装类型,可下载到D盘,后续自定义安装至C盘

 

  •  安装

安装包提取路径,应该是存储到临时文件,默认路径

 开始安装,选择自定义安装

 

 安装中取消勾选CUDA-Visual Studio Integration(集成至VS),由于还未安装visual studio,所以后续会提示未安装VS,,因此提前取消勾选

 电脑未安装VS提示

 

默认安装路径

 

 等待安装完成

 

  •  检查环境变量

检查是否自动添加成功了,包含环境变量,安装成功

 三、安装cuDNN

  • 环境要求

需要与cuda的版本对应,cudnn可以理解未cuda的一个深度学习插件,放到制动文件夹中无需安装

  • 下载

下载链接https://developer.nvidia.com/rdp/cudnn-archive

根据v11.5 cuda版本选择对应的cuDNN版本下载,需注册官网账号下载

  • 解压安装

解压

 安装

 将解压好的cuDNN与cuda根目录中对应好文件夹,将cuDNN中的文件复制拷贝过去

  •  配置环境变量

在path中新添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5libnvvp
  • 检查是否安装成功

进入cmd,输入如下指令

bandwidthTest是CUDA工具包中提供的一个示例程序。它旨在测量和显示以下几种内存传输的带宽:

  1. Host to Device : 从CPU(及其RAM)到GPU的数据传输速率
  2. Device to Host : 从GPU回到CPU的数据传输速率
  3. Device to Device : 在GPU内部,从一块内存区域传输到另一块内存区域的速率
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2
cd .\extras\demo_suite
.\bandwidthTest.exe

四、安装pytorch_gpu

环境要求

GPU版本安装需要显卡驱动,CUDA、cuDNN、python版本及pytorch各个版本需要对应兼容,并且在conda中新建一个虚拟环境安装pytorch

建立conda虚拟环境

进入conda prompt,输入以下指令:

conda create --name torenv python=3.8  # 创建python版本3.8的torenv环境
conda env list # 查看环境列表
conda activate torenv # 激活切换到torenv环境
conda deactivate   # 后续操作完成之后退出该环境

pytorch安装

pytorch版本列表查看符合安装要求的版本,cuda11.5在官网中未查询到,pytorch v1.12.0支持cuda11.3版本,cuda11.5可以使用v1.12.0,安装使用成功

五、换源下载

 可参清华开源软件镜像站说明

https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

豆源瓣,中国科技大学源,科学上网不换源速度是否可以?

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Total Commander的CudaLister插件(32位和64位版本),内含223个语法高亮文件。 Assembly.lcf Batch files.lcf C.lcf C#.lcf C++.lcf CSS.lcf Delphi resources.lcf Diff.lcf Go.lcf Haskell.lcf HTML.lcf Ini files.lcf JavaScript.lcf JSDoc.lcf JSON.lcf LESS.lcf Lua.lcf Makefile.lcf Markdown.lcf Pascal.lcf Perl.lcf PHP.lcf PowerShell.lcf Properties.lcf Python.lcf Ruby.lcf SQL.lcf XML.lcf TypeScript.lcf JavaScript Babel.lcf CoffeeScript.lcf Clojure.lcf Elixir.lcf Erlang.lcf MediaWiki.lcf Kotlin.lcf Racket.lcf GraphQL.lcf PlantUML.lcf LaTeX.lcf Caffe Prototxt.lcf Strace.lcf LiveCode.lcf Lisp.lcf Inno Setup.lcf Pascal script.lcf Twig.lcf Razor.lcf Kivy.lcf HTML Django DTL.lcf HTML Django DTL internal.lcf ZenScript (MineTweaker).lcf Stata.lcf Assembly SHARC DSP.lcf Assembly SPARC.lcf AviSynth.lcf HJSON.lcf MySQL SQL.lcf Assembly Z80 SjASM.lcf HTML Siteleaf Liquid.lcf HTML Siteleaf Liquid internal.lcf HTML Handlebars.lcf HTML Handlebars internal.lcf HTML Embedded JS internal.lcf Assembly AVR.lcf Apache Hive.lcf Squirrel.lcf Textile.lcf Lola-2.lcf PICL.lcf Rust.lcf Scala.lcf Objective-C.lcf MATLAB.lcf Java.lcf Graphviz DOT.lcf AutoIt.lcf AutoHotkey.lcf YAML.lcf Assembly MIPS.lcf Assembly ARM.lcf Bohemia SQF.lcf Intel HEX.lcf VimL.lcf Windows Resource Script.lcf Yacc.lcf reStructuredText.lcf ABC Notation.lcf Assembly JWASM.lcf RTF.lcf Modelica.lcf Jasmine JVM Assembler.lcf Java Velocity.lcf Bitsquid SJSON.lcf Dockerfile.lcf Oberon.lcf Abaqus Keywords.lcf CRF.lcf ColdFusion.lcf ARM Assembly.lcf PHP_.lcf HTML_.lcf Clavier.lcf NSIS.lcf Ragel.lcf Metafont.lcf AWK.lcf OpenSCAD.lcf Automake.lcf Vala.lcf Boo.lcf Bash script.lcf Nemerle.lcf R Markdown.lcf BibTeX.lcf OpenCL.lcf VBScript.lcf D.lcf Nim.lcf ABAP.lcf Visual dBase.lcf Visual Basic.lcf Prolog.lcf Asymptote.lcf Verilog HDL.lcf Pike.lcf MIPS Assembly.lcf Tcl.lcf Pawn.lcf Apache Pig.lcf HTML Smarty.lcf HTML Smarty internal.lcf Standard ML.lcf Great Cow Basic.lcf HTML Mustache.lcf HTML Diafan.lcf HTML Laravel Blade.lcf HTML Laravel Blade internal.lcf Tree.lcf CUDA C++.lcf Informix 4GL.lcf NFO files.lcf Text Russian detector.lcf GLSL.lcf Virgil.lcf Jinja2.lcf TOML.lcf Slim.lcf Haml.lcf Assembly FASM.lcf Gherkin.lcf Nginx.lcf Sass.lcf ActionScript.lcf Elm.lcf OpenEdge ABL.lcf MIB files (SNMP protocol).lcf Brainfuck.lcf Apache config.lcf Arduino.lcf Groovy.lcf F#.lcf FoxPro.lcf Puppet.lcf MSVS_Solution.lcf Swift.lcf ASAP2_database.lcf ASAP2_Meta_Language.lcf Julia.lcf AppleScript.lcf R.lcf Dart.lcf Scheme.lcf Fortran.lcf WSH_script.lcf CMake.lcf Stylus.lcf Jade.lcf Text_with_indentation.lcf PostScript.lcf Logfiles.lcf Haxe.lcf Caml.lcf Regular_expressions.lcf WinBuilder_script.lcf SQL_White.lcf SQL_Blue.lcf RPG_IV.lcf RPG_IV_(Free_Syntax).lcf PAX_Pascal.lcf Parser3.lcf OneC_(1C).lcf NSL_Assembler.lcf nnCron.lcf MySQL_Stored_Procedures.lcf MacroScript_(Macro_Sheduler).lcf JCL.lcf Gold_Parser.lcf Adobe_Flash.lcf XSLT.lcf VHDL.lcf T-SQL.lcf Smalltalk.lcf Rexx.lcf PL_SQL.lcf Modula 2.lcf KiXtart.lcf IDL language.lcf IDL files.lcf Harbour.lcf Graphviz.lcf GAMS.lcf Forth.lcf Factor.lcf Euphoria.lcf Eiffel.lcf CodeVisionAVR.lcf Cobol.lcf Clipper.lcf Clarion.lcf Ada (.ads).lcf Ada.lcf Acu Cobol.lcf
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值