首页 > 编程语言 >Python利用GPU进行深度学习

Python利用GPU进行深度学习

时间:2024-04-22 22:36:45浏览次数:24  
标签:训练 Python 深度 CUDA NVIDIA GPU 安装 环境变量

在深度学习当中,我们训练模型通常要对模型进行反复的优化训练,仅用CPU来进行训练的话需要花费很长时间,但是我们可以使用GPU来加速训练模型,这样就可以大大减少训练模型花费的时间。

 

首先我们需要一张NVIDIA显卡

  在搜索栏中搜索设备管理器

前往NVIDIA官网下载显卡对应的Studio驱动
  NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA

  (尽量安装比较早的驱动可以避免适配问题,新的应该也没什么大碍)

  下载完直接安装在默认目录就行了

打开CMD输入

  nvidia-smi

     获取驱动对应的CUDA版本

去CUDA官网下载对应的CUDA版本

CUDA Toolkit Archive | NVIDIA Developer  

同样安装在默认目录

 需要注意的是,我们在这里选择自定义安装方式,取消勾选Visual Studio Integration(如果你有Visual Stduio环境可以忽略)

其他的可以不管

安装成功后应当会在系统变量中自动添加这几个环境变量

 也可以打开cmd输入

  nvcc -V   进行检验

下载cudnn的包

 cuDNN Archive | NVIDIA Developer

解压之后,把这三个包复制到CUDA的本体文件夹中,然后替换相同文件,这个文件夹我们还需要设置环境变量

 

 在path中添加最后的环境变量

前四个都是CUDA本体,第五个是cuDNN这个包的bin文件夹(我把它放在CUDA本体的位置方便管理)

 

 

标签:训练,Python,深度,CUDA,NVIDIA,GPU,安装,环境变量
From: https://www.cnblogs.com/xxaxf/p/18151650

相关文章

  • Python企业面试题5 —— 网络编程和并发
    1.简述进程、线程和协程的区别以及应用场景?#进程:拥有自己独立的堆和栈,既不共享堆,也不共享栈,进程由操作系统调度。#线程:拥有自己独立的栈和共享的堆,线程也由操作系统调度。#协程和线程:协程避免了无意义的调度,由此可以提高性能;但同时协程失去了线程使用多CPU的能力。进程与......
  • 初中中考阅读理解难题一网打尽!句子结构深度解析+答案揭秘,助你轻松冲刺中考高分!-009
    PDF格式公众号回复关键字:ZKYDT009原文1Howdidthelotlookatthebeginningofthestory?解析1How怎么样did,thelot场地,look看起来,atthebeginningofthestory?在故事的开头故事开始时,那个场地看起来怎么样?2Thisplacelookslikeadump.这个地方看......
  • Mac搭建appium环境及python运行代码示例
    Appium主要是通过调用安卓提供的接口来执行命令的,所以需要安装Java和安卓SDK。1.安装Appium服务端appium的服务端是基于node的,直接使用npm(node包管理器)安装即可,比较简单。npminstall-gappium2.安装Python客户端pipinstallAppium-Python-Client同样直接使用pip安装......
  • python+appium+pytest做app自动化测试
    我在另一篇博客中写了使用unittest做app自动化测试的,包含了前期的环境的环境搭建,请参考如下链接:python+appium+unittest做app自动化测试这里,我们使用pytest框架再改写一个版本,因为pytest做测试报告看着更加好看,代码改良如下:fromappiumimportwebdriverimportpytest@pytest......
  • Python Numpy 矩阵运算
    目录1前言2点积与矩阵乘法2.1np.dot()2.2np.matmul()和@2.3np.multiply和*3矩阵的逆4Ref1前言Python中经常涉及到矩阵运算,其借助于Numpy库进行,因此本文记录一些基于Numpy的矩阵运算2点积与矩阵乘法矩阵的点积(dotproduct),又称为内积(innerproduct)$a=(x_1,y_1)......
  • python+appium+unittest做app自动化测试
    1.需要安装一些列的软件:(1)java(2)androidsdk:  https://www.cnblogs.com/chenxiaomeng/p/16544481.html(3)AppiumServerGUI(4)AppiumInspector (3和4老版本是一个)2.打开AppiumServerGUI直接点击startServer即可,使用默认配置 3.打开 AppiumInspectorremot......
  • python监控MongoDB服务进程,故障钉钉告警
     服务器1xx.168.8x.77#!/usr/bin/python#!_*_coding:utf-8_*_importosimportsysimporttimemongo_ip='192.168.xx.77'ports=['x001','x002']defport(ip,port):  response=os.popen("tcping %s%s|grepopen|awk-F'&......
  • 在Python中的for循环
    在Python中的for循环for循环:用于遍历序列(如列表、元组、字典、集合或字符串)或其他可迭代对象。pythonforiinrange(10):#这将循环10次,i的值从0到9print(i)while循环:当给定条件为真时,重复执行代码块。pythoni=0whilei<10:#这将循环10次print(i)i......
  • AI+X 经典共读-《动手学深度学习(Pytorch版)》Task打卡
    0.笔记仓库AI入门路线Task01:初识深度学习深度学习介绍和环境安装配置1.深度学习介绍1.1AI地图x轴:模式y轴:想做的东西感知:所见(人能够快速反应)推理:基于所见的想象知识:根据所见形成自己的知识规划:根据知识进行长远的规划自然语言处理:感知,用的最多的是机器翻译计算机视......
  • python 多并发
    多并发实现1#!/usr/bin/python2#-*-coding:utf-8-*-34"""5@File:.py6@Description:7@Time:2024/04/2210:00:008@Author:9@Version:1.010@Contact:[email protected]"""1213impor......