GPU版本PyTorch详细安装教程

2023-11-04

目录

一、安装显卡驱动

1、查看显卡驱动型号

2、下载显卡驱动

3、查看GPU状态

二、安装Visual Studio 2019

三、安装CUDA

1、下载对应版本的CUDA

2、安装下载好的CUDA

3、设置环境变量

 四、安装cudnn

五、安装anaconda

六、安装PyTorch

1、创建虚拟环境

2、激活并进入虚拟环境

3、安装PyTorch

4、验证PyTorch是否安装成功

注意:30系列的的显卡暂时不支持cuda11以下版本!!!

一、安装显卡驱动

1、查看显卡驱动型号

右击右下角开始,在设备管理器中查看计算机显卡型号,例如我的显卡是GTX1050:

2、下载显卡驱动

进入英伟达官网,下载对应的显卡驱动:官方驱动 | NVIDIAhttps://www.nvidia.cn/Download/index.aspx?lang=cn

选择对应版本参数并下载安装,安装时直接一直下一步即可。

3、查看GPU状态

安装完成后,打开CMD终端(快捷键:Win+R),命令行输入:nvidia-smi查看GPU状态:

 注意:右上角显示的CUDA版本(红色框中)是当前版本所支持的最高CUDA版本。

二、安装Visual Studio 2019

由于要使用cuda则必须要有一个编译工具,这里安装的是Visual Studio 2019

下载Visual Studio 社区版链接https://visualstudio.microsoft.com/zh-hans/downloads/

安装完成后进入以下界面,注意:勾选“Python开发“C++桌面开发

三、安装CUDA

1、下载对应版本的CUDA

 CUDA各版本官方下载地址: CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive

 CUDA用的是10.2版本,因此进入cuda官网找到对应的版本进行下载:

 

2、安装下载好的CUDA

下载完成后将CUDA安装,注意:建议默认路径,后期需要添加环境变量,安装时选择自定义安装

勾选Visual Studio Integration:

3、设置环境变量

鼠标右键计算机(此电脑),打开属性->高级系统设置->环境变量,可以看到系统变量中多了CUDA_PATHCUDA_PATH_V10_2两个环境变量。

接下来,还要在系统变量中“新建”,添加以下几个环境变量,以下是默认安装位置的路径的环境变量(左边是变量名,右边是变量值):

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2

CUDA_LIB_PATH = %CUDA_PATH%\lib\x64

CUDA_BIN_PATH = %CUDA_PATH%\bin

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

在系统变量 Path 的末尾添加:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;

双击Path,再添加如下5(默认安装路径):C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\bin\win64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2\common\lib\x64

 

四、安装cudnn

1、下载cuda10.2对应的cudnn

       cudnn下载地址https://developer.nvidia.com/cudnn

2、下载cudnn后直接将其解开压缩包,然后需要将解压后的bin,include,lib文件夹复制粘贴到cuda安装时的默认路径文件夹下(比如我的是在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 下)

注意:对整个文件夹bin,include,lib复制粘贴

3、最后测试cuda是否配置成功:

Win+R打开CMD终端执行:nvcc -V 即可看到cuda的信息:

五、安装anaconda

详细安装教程可参考以下链接:

(42条消息) Python安装教程步骤1:Pycharm和Anaconda的安装_ZHW_AI课题组的博客-CSDN博客https://blog.csdn.net/m0_37758063/article/details/116540837?spm=1001.2014.3001.5502

六、安装PyTorch

1、创建虚拟环境

在anaconda中添加镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

在开始栏打开刚才安装的anaconda prompt 创建项目运行虚拟环境:

conda create -n your_env_name(虚拟环境名称) python==xx(想要创建的虚拟环境的python版本号)

注:若不知道python版本,Win+R打开CMD终端执行:python,然后enter即可看到所安装的python版本号。

可以根据自己需要修改,例如:conda create -n mytorch python==3.9.7

输入y,然后按下enter确认开始下载安装。

出现以上界面则虚拟环境已经创建完成。

2、激活并进入虚拟环境

1、在上一步的基础上,对创建完成的虚拟环境进行激活,在anaconda prompt 终端中输入以下指令:

conda activate mytorch

2、若要退出当前的虚拟环境,执行以下指令即可:

conda deactivate

3、安装PyTorch

1、此时我们已经激活并进入到所创建的mytorch环境中。 

2、接下来进入PyTorch官网,选择相关参数,获取PyTorch安装指令,并在anaconda prompt 终端中执行红色框中的指令(该指令为最新版本的PyTorch)

注意:安装的时候要将命令后的-c pytorch后面的内容删除,从国内源进行下载,速度快一些。PyTorchhttps://pytorch.org/

另:若需要选择其他版本的PyTorch点击绿色框中的指令即可 

然后进入以下页面,找到需要的版本进行安装即可

3、输入PyTorch安装指令指令后,查看conda找到将要安装的包是否是要进行安装的

如果是,输入y之后,按下enter确认进行下载;

如果不是,就需要重新检查安装PyTorch的命令。

3、 安装PyTorch,至此,基础环境已经部署完成。

4、验证PyTorch是否安装成功

1、安装完成后,继续在在anaconda prompt 终端中执行以下指令,验证PyTorch是否安装成功

python 
import torch 
torch.cuda.is_available() 

2、如下图提示True说明框架配置成功,且GPU可用

 3、验证完成后Ctrl+Z回到命令行,然后执行conda list指令就可以看到该虚拟环境下已经安装好的包

能够在conda list中找到以上两个包,则说明已经配置完成。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

GPU版本PyTorch详细安装教程 的相关文章

随机推荐

  • 篇八:冒泡排序

    交换排序的一种 author xiaocui date 2022年08月02日 13 49 public class BubbleSort public static void main String args int arr new in
  • React 列表 & Keys

    React 列表 Keys const numbers 1 2 3 4 5 const listItems numbers map number gt li number li ReactDOM render ul listItems ul
  • leecode 题目5 最长回文子串(python)

    给你一个字符串 s 找到 s 中最长的回文子串 示例 1 输入 s babad 输出 bab 解释 aba 同样是符合题意的答案 示例 2 输入 s cbbd 输出 bb 代码是根据B站上的一位up主写的 使用动态规划算法 从小字符串衍生相
  • css3新弹性盒子-flex-grow

    目录 flex grow 1 flex grow定义和用法 2 flex grow属性值 flex grow说明 flex grow实例 flex grow flex grow属性在CSS新弹性盒模型中是比较常用 想要用好它就先了解它的 语
  • 花卉识别卷积神经网络

    卷积神经网络做的花卉识别 keras 五分类 向日葵 雏菊 郁金香 玫瑰 蒲公英 之后更怎么做的
  • 【专题5: 硬件设计】 之 【68.开关电源 之 buck电路中的电感电流波形】

    嵌入式工程师成长之路 系列文章 总目录 系列文章总目录 希望本是无所谓有 无所谓无的 这正如脚下的路 其实地上本没有路 走的人多了 也便成了路 原创不易 文章会持续更新 欢迎微信扫码关注公众号 承接 小程序 嵌入式 PC端项目开发 联系作者
  • Arduino基础入门篇31—LCD1602液晶显示

    LCD1602也叫1602字符型液晶 是一种专门用来显示字母 数字 符号的点阵型液晶模块 能同时显示16X2即32个字符 本篇我们来认识LCD1602 驱动它显示 Hello World 1 LCD1602介绍 对于单片机爱好者和电子爱好者
  • php-cgi.exe - FastCGI 进程超过了配置的请求超时时限

    解决方案一 处理 php cgi exe FastCGI 进程超过了配置的请求超时时限 的问题 内容转载 处理 php cgi exe FastCGI 进程超过了配置的请求超时时限 的问题 php技巧 脚本之家 详细错误 HTTP 错误 5
  • 【华为OD机试真题 JAVA】出错的或电路

    JS版 华为OD机试真题 JS 出错的或电路 标题 出错的或电路 时间限制 1秒 内存限制 262144K 语言限制 不限 某生产门电路的厂商发现某一批次的或门电路不稳定 具体现象为计算两个二进制数的或操作时 第一个二进制数中某两个比特位会
  • 利用Java对后端数据进行分页处理(Java假分页)

    代码如下 import java util Collections import java util List java分页 辅助器 public class PageHelper
  • HttpClient远程调用工具类

    文章目录 前言 一 HttpClient工具类 二 使用步骤 1 引入库 前言 提示 当前工具类get方式无法传分页数据 可自行篡改 不行就用post方式哈哈 提示 以下是本篇文章正文内容 下面案例可供参考 一 HttpClient工具类
  • Qt中显示摄像头数据(V4L2三)

    目录 一 通过V4l2来采集用qt显示 1 将V4L2获取摄像头数据封装成一个类 2 设置开始按钮开始采集数据 3 设置停止按钮停止采集数据 4 设置定时器不断采集数据 二 直接用qt多媒体模块来实现采集并且显示 1 在qt工程文件中添加
  • 【网络结构设计】7、RepVGG

    文章目录 一 背景 二 方法 2 1 使用简单结构的卷积神经网络的三个原因 2 2 使用多分支结构的训练时长 2 3 边端设备推理的重参数化 2 4 结构细节 三 效果 论文 RepVGG Making VGG style ConvNets
  • 解决野鸡大学水硕、水博?中国首次发布海外大学排名!

    本文分享中国首次发布的海外大学排名 来源 全国高校信息资料研究会与中国人民大学评价研究中心 新东方在线托福 青塔等平台 编辑 学妹 不久前 我国一科研机构首次发布了对海外学校的评估 网传 这或许会成为以后中国看待海归大学的背景考核依据之一
  • 深度学习:循环神经网络RNN及LSTM

    深度学习 循环神经网络RNN及LSTM 循环神经网络RNN 原理 代码 长短期记忆网络LSTM 原理 遗忘门 记忆门 输出门 代码 循环神经网络RNN 原理 对于传统的神经网络 它的信号流从输入层到输出层依次流过 同一层级的神经元之间 信号
  • 两行css代码实现瀑布流,html,css最简单的瀑布流实现方式且没有缺点!

    两行css代码实现瀑布流 html css最简单的瀑布流实现方式且没有缺点 之前一直使用css position定位 js原生的方法实现 代码如下 html代码段
  • 因果推断理论框架 Potenial Outcomes Framework

    1 Potenial Outcomes Framework 因果效应通常无法直接计算 无法同时观测一个样本施加和不施加干预的结果 所以通常是通过观测数据推断 由于相关性 neq 因果性 观测结果不直接等于ATE Potenial Outco
  • SpringBoot 基本使用

    目录 热部署 方式一 Spring Boot DevTools 方式二 使用JRebel插件 web静态资源的存放 json日期时间设置 组件扫描 资源导入 运行项目的多个实例 资源文件的拷贝问题 读取resources下的文件 SpEL的
  • python三本经典书籍-关于 Python 的经典入门书籍有哪些?

    先说明 这些书是我整理来准备以后慢慢看的 并没有全部购买 更没有全部阅读 排名不分先后 Learn Python the Hard Way 本书是一本Python入门书籍 适合对计算机了解不多 没有学过编程 但对编程感兴趣的读者学习使用 这
  • GPU版本PyTorch详细安装教程

    目录 一 安装显卡驱动 1 查看显卡驱动型号 2 下载显卡驱动 3 查看GPU状态 二 安装Visual Studio 2019 三 安装CUDA 1 下载对应版本的CUDA 2 安装下载好的CUDA 3 设置环境变量 四 安装cudnn