Ubuntu22.04安装PyTorch1.12.1 GPU版本

news2024/12/24 21:17:25

目录

一、电脑相关信息

1. 电脑显卡环境:

二、安装Pytorch1.12.1/cu116(GPU版本)

1. 准备:新建虚拟环境

2. 方式一:用pip在线安装torch1.12.1+cu116、torchvision0.13.1+cu116:(在pytorch官网的历史版本里找安装命令)

3. 方式二:用conda在线安装pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.6:(在pytorch官网的历史版本里找安装命令)


最近又被装环境折磨,仅根据实际情况记录糟心经历。仅作记录。

网上很多pytorch GPU版本安装教程里一般步骤都说是:装显卡驱动、装cuda、装cudnn、最后安装pytorch GPU环境。但也有教程说可以不单独安装cuda、cudnn,直接安装pytorch GPU环境。于是,我打算在不安装cuda、cudnn的情况下(主要是嫌安装太麻烦),直接安装pytorch GPU环境(最后安装好是pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.6)。

机器情况:新机器,自带的有nvidia-smi显卡驱动。

机器:Ubuntu22.04 + 英伟达显卡(GeForce RTX 4090)+ Anaconda3。

目标:安装gpu版本的PyTorch1.12.1(torch1.12.1/cu116 + torchvision0.13.1/cu116)

一、电脑相关信息

1. 电脑显卡环境:

  • 显卡:Nvidia GeForce RTX 4090
  • 电脑的显卡驱动版本:525.105.17,CUDA Version:12.0

(CUDA Version12.0表示此显卡驱动最高支持cu12.0。)

二、安装Pytorch1.12.1/cu116(GPU版本)

1. 准备:新建虚拟环境

1)首先安装Anaconda。(很简单,自行搜索)

2)其次在用Anaconda建一个虚拟环境。

conda create -n torch1-12-1 python=3.10
#(torch1-12-1 表示自己起的环境的名字。)

3)激活虚拟环境。(后续的操作都是在该虚拟环境下进行的。)

conda activate torch1-12-1

2. 方式一:用pip在线安装torch1.12.1+cu116、torchvision0.13.1+cu116:(在pytorch官网的历史版本里找安装命令)

1)在PyTorch官网历史版本链接( Previous PyTorch Versions | PyTorch)里面,找到使用pip安装pytorch1.12.1/cu116、torchvision==0.13.1+cu116的安装命令,进行在线安装。

安装命令:

# CUDA 11.6
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

安装很快,很快就安装好了。测试torch的版本以及torch.cuda.is_available()都没有问题。

但是,在用这个pytorch环境运行代码时,总是报关于cuda的错(此代码在别的老机器上没有问题)。最后感觉可能是因为pip命令安装的时候没有顺带安装cuda,再加上自己电脑自己没有安装cuda,所以感觉是环境没有安装好然后运行代码就一直报错。

于是选择用下面的方式二重新安装pytorch环境。

3. 方式二:用conda在线安装pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.6:(在pytorch官网的历史版本里找安装命令)

最后选择用conda命令安装pytorch GPU版本。因为conda命令里面带的有cudatoolkit。

1)在PyTorch官网历史版本链接( Previous PyTorch Versions | PyTorch)里面,找到使用conda安装pytorch1.12.1/cu116、torchvision==0.13.1+cu116的安装命令,进行在线安装。

安装命令:

# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

但是用conda在线安装pytorch环境的时候,下载包的时候老是断掉,下了一点就断了报错了。来回重新运行conda安装的命令很多次,一直没有下载安装成功。

最后配了一下conda的镜像源(照着这个链接里面配的anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror),不知道是因为这个原因还是突然网速变快了,突然一下子下载得贼快,几分钟就安装好了。真的是有点玄学。。。。

如下图:(下载安装得很快)

测试:

使用下面的命令测试,可以看到torch和torchvision对应的版本。

(base) ubuntu@ubuntu:~$ conda activate torch1-12-1
(torch1-12-1) ubuntu@ubuntu:~$ python
Python 3.10.11 (main, Apr 20 2023, 19:02:41) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> torch.__version__
'1.12.1'
>>> torchvision.__version__
'0.13.1'
>>> torch.cuda.is_available()
True
>>> torch.version.cuda
'11.6'
>>> torch.backends.cudnn.version()
8302
>>> exit()
(torch1-12-1) ubuntu@ubuntu:~$ 

PyTorch1.12.1 GPU版本(pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6)就安装好了~ 可以愉快使用了~

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

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

相关文章

堆的实现,以及画图和代码分析,建堆,堆排序,时间复杂度的分析

堆的实现 堆的概念及结构堆的实现初始化销毁返回堆顶元素判空有效数据个数 堆的插入(向上调整算法)删除堆顶元素,仍然保持堆的形态(向下调整算法)堆的创建向上调整法建堆向下调整建堆两种建堆方法时间复杂度向下调整法…

shell脚本之数值,冒泡排序算法

目录 一、数组 1.定义数组 2. 数组中数据类型 2.1数值类型 2.2字符类型 二、数组的用法 1. 输出数组中的值 2. 统计数组参数个数 ​编辑 3.查看数组下标列表 4.分割字符串 5.替换数组中的字符 6.删除数组 三、数组追加元素 1.方法1示例 2.方法2示例 3.方法3示例 …

Notion Ai中文指令使用技巧

Notion AI 是一种智能技术,可以自动处理大量数据,并从中提取有用的信息。它能够 智能搜索:通过搜索文本和查询结果进行快速访问 自动归档:可以根据关键字和日期自动将内容归档 内容分类:可以根据内容的标签和内容的…

【嵌入式Linux】MBR分区表 和 GPT分区表

文章目录 GUID以及分区表MBR分区方案GPT 分区方案GPT分区表结构 GPT分区表LBALBA0LBA1LBA 2-33python生成GPT分区表gpt分区表实例 gpt分区表查看查看百问网T113-s3固件查看友善之臂nanopi-m1-plus官方固件查看荣品RV1126固件查看f1c200s固件查看V3s的SD启动卡 原文:…

session和Filter

session 介绍 jsp利用servlet提供的HttpSession接口识别一个用户,存储这个用户的所有访问信息。默认情况下,jsp允许会话跟踪,一个新的HttpSession对象将会自动地为新的客户端实例化。禁止会话跟踪需要关掉它,通过将page指令中ses…

多模态:BLIP-2论文讲解

多模态:BLIP-2论文讲解 IntroductionMethod第一阶段第二阶段 实验 Introduction 多模态学习在近两年我们已经见证了他的快速发展,由于它是视觉-语言的交叉领域,我们自然地期待可以借助目前风头正盛的LLM来辅助完成多模态任务。 在这篇论文中…

【SAP Abap】X-DOC:SNRO - ABAP流水号应用

【SAP Abap】X-DOC:SNRO - ABAP流水号应用 1、定义表(字段域)2、定义流水号3、使用流水号4、测试程序 1、定义表(字段域) 2、定义流水号 Tcode: SNRO/SNUM, 根据以上创建的字段域 YDSNRO,创建对…

grep练习题

题目: 文件格式: 第1列:城市位置编号。 第2列:月份。 第3列:存储代码及出库年份。 第4列:产品代号。 第5列:产品统一标价。 第6列:标识号。 第7列:合格数量。 file.txt文件内容&a…

RabbitMQ入门案例及简单工程搭建

环境的搭建 这里是用Maven工程搭建的基础环境项目&#xff0c;这里的dome_rabbitmq就是父工程。 子工程 publisher&#xff1a;消息发布者&#xff0c;将消息发送到队列queueconsumer&#xff1a;订阅队列&#xff0c;处理队列中的消息 父工程的pom文件依赖 <?xml ver…

SpringCloud-11_Alibaba Sentinel

SpringCloud系列 SpringCloud-10_Alibaba Nacos SpringCloud-9、SleuthZipkinSpringCloud-8、Gateway网关服务SpringCloud-7_OpenFeign服务调用SpringCloud-6_Ribbon负载均衡SpringCloud-5_模块集群化 求帮助 富文本编辑器用着真伤心TT。。不知道编辑器发生了什么。。--- 用…

Android Studio Notification(状态栏通知) 不显示通知 解决

引言&#xff1a;在学习的过程中&#xff0c;我发现&#xff0c;无论怎么修改&#xff0c;甚至是直接复制了大佬的代码&#xff0c;我的程序都不呢个正确的弹出一个状态栏通知&#xff0c;在经过一晚上的纠缠后&#xff0c;终于找到了原因 通知不显示可能由多种原因引起&#…

C++自动推导与函数模板

自动推导、函数模板、类模板 目录 1. 自动推导出数据类型 2. 函数模板 基本概念注意事项函数模板的具体化函数模板分文件编写函数模板高级函数后置返回类型 1. 自动推导出数据类型 auto关键字 linux 系统下使用的话&#xff0c;要在编译时 —stdc11 注意&#xff1a; a…

关于B+树的介绍、用途和c++代码实现

数据结构和算法的重要性不言而喻&#xff0c;一些优秀的开源项目的核心和灵魂就是数据结构、算法。在实际的编程中我们经常可以在各种框架、算法中看见B树、B树的身影。特别是在数据库的数据库引擎中&#xff0c;它们更是占据着重要的地位。 下面我将通过简单的二叉树&#xff…

2023/5/12总结

这俩天主要花时间在项目上&#xff1a; 实现了创建群聊和添加群聊&#xff1a; 点击创建群聊&#xff1a; &#xff0c;点击确定之后&#xff0c;会分配到一个群聊&#xff0c;默认头像会是下面这个圆形的头像&#xff1a; 添加群聊&#xff1a; 如果你要加入的群聊在自己的列…

06-HTML-列表标签

1、 <ul> 标签定义无序列表。 2、<ol> 标签定义有序列表。 属性值描述compactcompact HTML5 中不支持。HTML 4.01 中不赞成使用。 规定列表呈现的效果比正常情况更小巧。 reversedreversed规定列表顺序为降序。(9,8,7...)startnumber规定有序列表的起始值。type 1A…

为什么使用ConcurrentHashMap

currentHashMap的介绍 ConcurrentHashMap是线程安全并且高效的一种容器,我们就需要研究一下ConcurrentHashMap为什么既能够保证线程安全,又可以保证高效的操作。 为什么使用ConcurrentHashMap,我们就需要和HashMap以及HashTable进行比较&#xff1f; HashMap是线程不安全的&…

唐朔飞计组 第六章运算方法简单复习

在计算机中参与运算的数有两类&#xff1a;有符号数和无符号数 int 和unsigned unsigned可以看成是正数或者绝对值。 有符号数分为原码反码和补码 原码和反码的表示范围是相同的 但是补码由于将-0的位置换成2^n所以补码表示范围比原码和反码要多一位&#xff0c; 判断溢出比较…

诚邀社区开发者参与DeepBook测试和集成

DeepBook是Sui的基础流动性层&#xff0c;Sui基金会诚挚邀请社区开发者参与其测试和集成。 DeepBook为Sui的原生中央订单簿&#xff08;Central Limit Order Book&#xff0c;CLOB&#xff09;和基础流动性层&#xff0c;将会在未来数周准备完成&#xff0c;我们邀请大家参与测…

Unity大面积草地渲染——4、对大面积草地进行区域剔除和显示等级设置

目录 1、Shader控制一棵草的渲染 2、草地的动态交互 3、使用GPUInstancing渲染大面积的草 4、对大面积草地进行区域剔除和显示等级设置 Unity使用GPU Instancing制作大面积草地效果 大家好&#xff0c;我是阿赵。 这里开始讲大面积草地渲染的第四个部分&#xff0c;对大面积草地…

零知识证明:安全定义

之前在本科的课程仅仅略微介绍了下零知识证明&#xff0c;之后自学了一些相关内容&#xff0c;但不成体系。本学期跟着邓老师较为系统地学习了 ZKP&#xff0c;发现自己之前有很多的误解&#xff0c;临近期末整理下重要内容。 参考文献&#xff1a; Goldreich O. Foundations…