Pytorch GPU版本安装

一、背景

       记录一下安装Pytorch GPU版本过程。

       由于手残,卸载了电脑上的显卡驱动,现在我连显卡类型是啥都不知道了。

       总体思路:安装显卡驱动->安装cuda->安装pytorch库

二、安装显卡驱动

2.1 查看本地显卡型号

        通过「DirectX 诊断工具」查看显卡信息,以便下载驱动程序。

  键盘win+r  输入 dxdiag 进入诊断工具界面

2.2 下载显卡驱动程序

        进入人工智能计算领域的领导者 | NVIDIA英伟达官网,点击驱动程序。

        根据自己电脑配置信息进行选择,然后点击搜索->下载。

        下面进行安装,同意并继续->下一步->马上重启

        cmd输入 nvidia-smi 可以看到驱动安装成功。注意下面驱动版本是552.22,后面的CUDA 版本是当前驱动支持的最高可以安装的版本

三、安装CUDA

3.1 判断所需CUDA版本       

 根据实际情况,我需要安装pytorch版本为torch==1.13.1。因为cuda、pytorch版本之间是有依赖性的,因此需要选择适合我们的CUDA版本。见官网Previous PyTorch Versions | PyTorch。

       找到了我需要的CUDA版本11.6或11.7。同时,此CUDA版本符合驱动能支持的最高CUDA版本,否则需要重新安装驱动。

       接下来,去官网下载CUDA11.7CUDA Toolkit Archive | NVIDIA Developer。

3.2 安装所需CUDA版本

        检查是否安装成功,cmd中输入 nvcc -V

四、CUDNN安装(可选)

        参考这篇文章GPU,CUDA,cuDNN的理解-CSDN博客,文章说CUDNN不是必须安装的。

        我这里选择不安装。CUDNN下载地址cuDNN Archive | NVIDIA Developer。注意这个网站需

要注册(无语凝噎)

五、安装pytorch库

        回到Previous PyTorch Versions | PyTorch选择自己要下载的torch版本。

        我直接复制这个命令进行安装。

        下载完成后验证。命令行输入  conda list

        最后测试pytorch是否可以调用GPU,大功告成。

# -*- coding:utf-8 -*-
# @FileName :test_GPU.py
# @Time :2024/4/28 14:21
# @Author :Tim
import torch
if torch.cuda.is_available():print("GPU is available")
else:print("GPU is not available")

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/5559.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Leetcode 145:二叉树的后序遍历(迭代法)

给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。 思路: 迭代法的思路是,使用栈,一层一层的将树节点遍历入栈。 比如下面这个树,使用迭代法,1)第一层,让根节点入栈。2&a…

顺序表??

1.引入:首先需要了解数据结构的作用:是计算机存储、组织数据的方式。就比如之前学的数组,就是一种基础的数据结构。通过数组,就可以去维护很多个同类型的数据,实现增加删除等的功能,这样就可以有效管理这些…

Java中的ArrayList、List、[]、Arrays等都是个啥

在Java开发中那些场景推荐使用那些类? ArrayList: 当需要一个动态大小的数组,可以动态增长和缩减时,推荐使用 ArrayList。它提供了高效的随机访问和常数时间的插入/删除操作。 Arrays 类: 当需要对数组进行排序、搜索或…

java例子一

在Java中实现去水印视频的功能通常涉及到视频处理,这可能需要使用一些专门的库来解码、处理和重新编码视频。一个常用的库是Xuggler,它是一个基于FFmpeg的Java库,可以用来处理音频和视频数据。但是需要注意的是,Xuggler项目已经不…

嵌入式学习63-C++

知识零碎: newmalloc friend 友元函数 …

微搭低代码入门01微搭底层能力介绍

目录 1 什么是应用2 有服务器和无服务器3 关系型数据库和文档型数据库4 云存储5 云函数6 静态网站托管总结 日常经常解答初学者的问题,问的最多的就是学习微搭,需要多久,要不要学习代码。有的初学者心比较急躁,总是希望一蹴而就&a…

边循环边删除List中的数据

List边循环,边删除;这种一听感觉就像是会出问题一样,其实只要是删除特定数据,就不会出问题,你如果直接循环删除所有数据,那可能就会出问题了,比如: public static void main(String[…

一些优雅的监控运维技巧

准备工作 安装 sysstat sudo apt install sysstat查看某个进程的cpu情况 pidstst -u -p 256432查看某个进程的RAM情况 pidstst -r -p 256432查看某个进程的IO情况 pidstst -d -p 256432查看某个进程下的线程执行情况 pidstst -t -p 256432查看指定PID的进程对应的可执行文件…

读写锁精讲:Java中使用ReadWriteLock提升性能的终极指南

1. 读写锁基础 1.1 什么是ReadWriteLock 在并发编程中,ReadWriteLock是一个锁,它允许多个线程同时读共享数据,而写操作则是互斥的。这意味着如果没有线程正在对数据进行写入,那么多个线程可以同时进行读取操作,从而提…

C++协程库封装

操作系统&#xff1a;ubuntu20.04LTS 头文件&#xff1a;<ucontext.h> 什么是协程 协程可以看作轻量级线程&#xff0c;相比于线程&#xff0c;协程的调度完全由用户控制。可以理解为程序员可以暂停执行或恢复执行的函数。将每个线程看作是一个子程序&#xff0c;或者…

OpenCV如何实现背投(58)

返回:OpenCV系列文章目录&#xff08;持续更新中......&#xff09; 上一篇&#xff1a;OpenCV直方图比较(57) 下一篇&#xff1a;OpenCV如何模板匹配(59) 目标 在本教程中&#xff0c;您将学习&#xff1a; 什么是背投以及它为什么有用如何使用 OpenCV 函数 cv::calcBackP…

蓝桥杯国赛填空题(弹珠堆放)

小蓝有 20230610 颗磁力弹珠&#xff0c;他对金字塔形状尤其感兴趣&#xff0c;如下图所示&#xff1a; 高度为 1 的金字塔需要 1 颗弹珠&#xff1b; 高度为 2 的金字塔需要 4 颗弹珠&#xff1b; 高度为 3 的金字塔需要 10 颗弹珠&#xff1b; 高度为 4 的金字塔需…

Python 语音识别系列-实战学习-语音识别特征提取

Python 语音识别系列-实战学习-语音识别特征提取 前言1.预加重、分帧和加窗2.提取特征3.可视化特征4.总结 前言 语音识别特征提取是语音处理中的一个重要环节&#xff0c;其主要任务是将连续的时域语音信号转换为连续的特征向量&#xff0c;以便于后续的语音识别和语音处理任务…

qt嵌入并控制外部程序

一、流程 1、调用Window接口模拟鼠标&#xff0c;键盘事件 POINT point; LPPOINT lpppoint &point; GetCursorPos(lpppoint);//获取鼠标位置 SetCursorPos(point.x, point.y);//设置鼠标位置//鼠标左键按下 mouse_event(MOUSEEVENTF_LEFTDOWN | MOUSEEVENTF_LEFTUP, poi…

逻辑漏洞:初识水平越权与垂直越权

目录 1、什么是越权漏洞呢&#xff1f; 2、水平越权 3、垂直越权 4、burpsuite autorize插件 最近在学习逻辑漏洞的相关知识和技能&#xff0c;这里pikachu靶场作为演示进行学习一下&#xff1a; pikachu靶场&#xff1a;GitHub - zhuifengshaonianhanlu/pikachu: 一个好玩…

计算机服务器中了devicdata勒索病毒怎么办?Devicdata勒索病毒解密工具步骤

在这个网络飞速发展的时代&#xff0c;网络为企业的生产运营起到了关键性作用&#xff0c;利用网络可以开展各项工作业务&#xff0c;大大提高了企业生产效率与业务水平&#xff0c;在大家都为网络的便利感到欣慰时&#xff0c;网络数据安全问题&#xff0c;成为众多企业关心的…

go语言获取变量类型的4种方式

在go语言中我们常常需要获取某个变量的类型&#xff0c;其他语言如python可以使用 type(x), javascript中可以使用 typeof x 获取变量类型&#xff0c; Go 语言中我们也可以通过一下4种方式获取变量的类型。 1. 通过 fmt.Printf 的 %T 打印变量的类型&#xff1b; var x flo…

std::enable_shared_from_this 有什么意义?

问&#xff1a; 这是boost里面举的一个例子&#xff1a; class Y: public enable_shared_from_this<Y> { public:shared_ptr<Y> f(){return shared_from_this();} }int main() {shared_ptr<Y> p(new Y);shared_ptr<Y> q p->f();assert(p q);asser…

快手客户端一二面+美团前端一面+腾讯企业微信开发客户端一面

快手一面结志 1、自我介绍 2、对称加密非对称加密 3、TCP/UDP 4、在学校有什么课程是强项&#xff0c;说了过去几次面试中面到的C的语言基础知识 5、问C、Java中兴趣在哪里 6、问到项目&#xff0c;自己做的还是跟着学校老师做的&#xff0c;同样问到兴趣在哪里 7、LRU …

模型智能体开发之metagpt-多智能体实践

参考&#xff1a; metagpt环境配置参考模型智能体开发之metagpt-单智能体实践 需求分析 之前有过单智能体的测试case&#xff0c;但是现实生活场景是很复杂的&#xff0c;所以单智能体远远不能满足我们的诉求&#xff0c;所以仍然还需要了解多智能体的实现。通过多个role对动…